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

Question

我正在使用 pandas 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

一种方法是分组并应用函数获取列表，然后转换为字典。

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