Pandas字典转换为字典

Question

我有这个数据框

mylist = [['a', 1], ['a', 2], ['b', 3], ['b', 4]]
df = pd.DataFrame(mylist)

我想把它变成这个

desired_dict = {'a':[1, 2], 'b':[3, 4]}

使用 pandas 的优雅方法是什么？

Answer 1

尝试：

df.groupby(0)[1].agg(list).to_dict()

Answer 2

你可以使用

out = df.groupby(0)[1].apply(list).to_dict()

print(out)

{'a': [1, 2], 'b': [3, 4]}