如何选择列值大熊猫显示GROUPBY

Question

我使用熊猫0.16.0，我有数据：

id A B 
1 10 100 
2 10 101 
3 20 102 

当我打电话

df.groupby(['A']).groups 

我

{10: [1 2], 20: [3]} 

我想要这个（来自B列的值）

{10: [100, 101], 20: [102]} 

请帮助

Answer 1

一种方式是GROUPBY和应用功能，采取列表，然后转换成字典。

In [92]: df.groupby(['A']).apply(lambda x: x['B'].tolist()).to_dict() 
Out[92]: {10: [100, 101], 20: [102]}