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

Question

我正在使用pandas 0.16.0，我有数据：

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

当我打电话

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

我有

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

我想要这个（B列的值）

sigint-ignore

请帮助

Answer 1

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

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