python - 长格式 Pandas 数据框到字典

Question

虽然我找到了有关如何将 pandas DataFrame 转换为字典的帮助和文档，以便列是键，值是行，但当我想将列的值之一作为键并将相关值从另一列作为值，以便像这样的 df

a b
1 car
1 train
2 boot
2 computer
2 lipstick

转换为以下字典{'1': ['car','train'], '2': ['boot','computer','lipstick]}

我觉得这很简单，但我没有想法。我尝试了 df.groupby('a').to_dict() 但没有成功

有什么建议吗？

Answer 1

您可以将其视为分组聚合(即，将每个组转换为一个值的操作——在本例中为列表):

In [85]: df.groupby(['a'])['b'].agg(lambda grp: list(grp))
Out[85]: 
a
1                  [car, train]
2    [boot, computer, lipstick]
dtype: object

In [68]: df.groupby(['a'])['b'].agg(lambda grp: list(grp)).to_dict()
Out[68]: {1: ['car', 'train'], 2: ['boot', 'computer', 'lipstick']}