私はパンダを使用してDATAFRAME持っているのpython

Question

でGROUPBYパンダからベクトルを抽出する方法：今

one two three 

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

を、私は「3」をグループ化することによってベクトルを抽出します。 基本的に、私は「3」をグループ化に基づいて「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]] 

もう1つの解決策：

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]]