Pandas 数据框对多列和要列出的值进行 dict

Question

我有一个数据框

id    key
a1     1
a2     1
a3     1
a4     2
a5     2
a6     3

我想创建一个字典，以key 作为机器号，id 列作为列表

喜欢：

{1: ['a1', 'a2', 'a3'], 2: ['a4', 'a5'], 3: ['a6']}

我可以先用 groupby 再做 .to_dict 吗？

Answer 1

我相信您需要 dict 的列表广告值 - 使用 groupby + apply + to_dict：

d = df.groupby('key')['id'].apply(list).to_dict()
print (d)
{1: ['a1', 'a2', 'a3'], 2: ['a4', 'a5'], 3: ['a6']}

或者如果需要带有标量的list，请将if/else 添加到apply：

d = df.groupby('key')['id'].apply(lambda x: list(x) if len(x) > 1 else x.iat[0]).to_dict()
print (d)
{1: ['a1', 'a2', 'a3'], 2: ['a4', 'a5'], 3: 'a6'}

Answer 2

在 groupby 迭代器周围使用字典理解

{n: v.tolist() for n, v in df.groupby('key').id}