如何从python中的groupby pandas中提取一个向量

Question

我有一个使用pandas的DataFrame：

one    two  three

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

现在，我将通过分组“三个”来提取a向量。 基本上，我应该从“＆＃;基于分组的列＆＃34;三＆＃34;：

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