如何选择要在pandas groupby中显示的列值

时间:2021-07-28 09:07:19

I am using pandas 0.16.0, I have data:

我使用的是pandas 0.16.0,我有数据:

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

when I call

我打电话的时候

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

I have

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

and I want to have this (values from column B)

我想要这个(B栏的值)

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

please help

1 个解决方案

#1


One way is to groupby and apply function to take list, and then convert to dict.

一种方法是groupby并将函数应用于列表,然后转换为dict。

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

#1


One way is to groupby and apply function to take list, and then convert to dict.

一种方法是groupby并将函数应用于列表,然后转换为dict。

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