将 Python Dataframe 中的项目存入字典

Question

假设我有一个 Python 数据框：

Column A	Column B
A	Val 1
A	Val 2
B	Val A
B	Val B
B	Val C
B	Val D

我想将 B 列存入一个字典，其中键 = A 列的唯一值，如下所示：

out = { 'A': ['Val 1','Val 2'], 'B': ['Val A','Val B','Val C','Val D']}

我该怎么做？

我尝试制作一个数据透视表，但它只允许聚合 B 列；我希望它们作为列表中的单独值。

Answer 1

一种使用pandas.DataFrame.groupby的方法：

out = df.groupby("Column A")["Column B"].apply(list).to_dict()

输出：

{'A': ['Val 1', 'Val 2'], 'B': ['Val A', 'Val B', 'Val C', 'Val D']}