如何選擇列值大熊貓顯示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]}