基于数据帧的列创建字典

Question

我有一个数据帧的形式：

pandas as pd
df = pd.DataFrame({'col1': [1,2,3,4,5,6], 
                   'col2': ['A','A','B','C','D','D']}, 
                   index=['row1', 'row2','row3', 'row4','row5', 'row6'])

我想基于dataframe df的值创建以下字典：

dict = {'A':[1,2], 'B':[3], 'C':[4], 'D':[5,6]}

Answer 1

按^{}将每个组的col1值转换为列表，然后调用^{}：

d = df.groupby('col2')['col1'].apply(list).to_dict()
print (d)
{'A': [1, 2], 'B': [3], 'C': [4], 'D': [5, 6]}