用唯一的键和值列表从两列中创建一个字典

pyl*_*ner 3 python dictionary dataframe pandas

我正在尝试创建一个具有唯一键和多个值的字典

DF :

key value
2   21
2   32
2   455
3   12
3   45
3   21
Run Code Online (Sandbox Code Playgroud)

预期输出:

{'2' : ['21', '32', '455'], '3': ['12','45','21']}
Run Code Online (Sandbox Code Playgroud)

代码 :

dict(zip(df['key'], df['value']))
Run Code Online (Sandbox Code Playgroud)

需要一些帮助

jez*_*ael 5

首先聚合列表 by GroupBy.agg,然后转换Series为字典 by Series.to_dict

d = df.groupby('key')['value'].agg(list).to_dict()
print (d)
{2: [21, 32, 455], 3: [12, 45, 21]}
Run Code Online (Sandbox Code Playgroud)