熊猫词典向词典的转换

Question

我有这个数据

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

我想把它变成这样

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

对熊猫来说，什么是一种优雅的方法呢？

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]}