python - 从没有循环的数据框创建字典

Question

下面是我的数据框示例和我希望从中计算的内容的字典。我当前的方法包括遍历唯一的 ColA 值，创建一个子集数据框，获取 ColB 值的列表，并从中制作一个 dict。问题是我有超过一百万个独特的 ColA 值要循环。有任何想法吗？？

东风

 ColA       ColB
 mike        34
 mike         3
 mike        10
 bill        80
 dean         2
 dean         4
 dean        44
 dean        56

想要的字典 ={'mike':[34,3,10], 'bill': [10], 'dean': [2,4,44,56]}

有任何想法吗？谢谢！！

标签： pythonpandasdataframe

Answer 1

out = df.groupby("ColA")["ColB"].agg(list).to_dict()
print(out)

印刷：

{'bill': [80], 'dean': [2, 4, 44, 56], 'mike': [34, 3, 10]}