将pandas df转换为基于重复行的字典(row必须是键)

Question

我想让hdd转换成dict，如下所示:我有这样的df:

col1   col2
"a"   12
"a"   2
"b"   34
"c"   9
"c"   45

，我需要这样的字典:

d = {"a":[12, 2], "b":[34], "c":[9, 45]}

谁能给我一些建议?请帮助。

Answer 1

先聚合到lists，再转换输出到dict:

df.groupby('col1')['col2'].agg(list).to_dict()