python - Pandas groupby 两列然后获取值的字典

我有一个 Pandas 数据框:

banned_titles =
TitleId  RelatedTitleId
0    89989           32598
1    89989         3085083
2    95281         3085083

当我按如下方式申请 groupby 时

In [84]: banned_titles.groupby('TitleId').groups
Out[84]: {89989: [0, 1], 95281: [2]}

这是如此接近，但不是我想要的。

我想要的是:

{89989: [32598, 3085083], 95281: [3085083]}

有没有办法做到这一点？

最佳答案

试试这个:

In [8]: x.groupby('TitleId')['RelatedTitleId'].apply(lambda x: x.tolist()).to_dict()
Out[8]: {89989: [32598, 3085083], 95281: [3085083]}

或作为一系列列表:

In [10]: x.groupby('TitleId')['RelatedTitleId'].apply(lambda x: x.tolist())
Out[10]:
TitleId
89989    [32598, 3085083]
95281           [3085083]
Name: RelatedTitleId, dtype: object

数据:

In [9]: x
Out[9]:
   TitleId  RelatedTitleId
0    89989           32598
1    89989         3085083
2    95281         3085083

关于python - Pandas groupby 两列然后获取值的字典，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/37690092/