如何提取GROUPBY大熊貓蟒蛇

Question

矢量我有使用熊貓一個數據幀：

one two three 

1  2 1 
4  1 1 
2  2 1 
3  1 2 
20  2 2 

現在，我將通過分組「三」提取的向量。 基本上，我應該基於分組「三」從「兩」列得到向量：

groupby('three') 
a=[2,1,2] 
b=[1,2] 

非常感謝

Answer 1

您可以使用groupby：

s = df.groupby('three')['two'].apply(list) 
print (s) 
three 
1 [2, 1, 2] 
2  [1, 2] 
Name: two, dtype: object 

a = s.loc[1] 
b = s.loc[2] 
print (a) 
[2, 1, 2] 

print (b) 
[1, 2] 

如果需要嵌套的列表：

L = df.groupby('three')['two'].apply(list).tolist() 
print (L) 
[[2, 1, 2], [1, 2]] 

另一種可能的解決方案：

L = [list(x) for i, x in df.groupby('three')['two']] 
print (L) 
[[2, 1, 2], [1, 2]] 

L = [x.tolist() for i, x in tuple(df.groupby('three')['two'])] 
print (L) 
[[2, 1, 2], [1, 2]]