python - Pandas Dataframe 到带有元组的列表字典

Question

我在 Dataframe 中有一些数据(例如，df):

Col_1   Col_2   Col_3
Key1    ACURA   CL
Key2    ACURA   EL
Key2    ACURA   ILX
Key3    ACURA   INTEGRA
Key4    ACURA   INTEGRA

我希望看到的方式:

{'key1': [('ACURA', 'CL')], 
 'key2': [('ACURA', 'EL'), ('ACURA', 'ILX')], 
 'key3': [('ACURA', 'INTEGRA')], 
 'key4': [('ACURA', 'INTEGRA')]}

我尝试了类似 df.set_index('Col_1').T.to_dict('list') 的操作，但失败了，因为 Col_1 具有非唯一值。 当然，有一个简单的解决方案可以逐行运行 df 并手动创建我的字典，但我更喜欢更 Pythonic 的方式。

Answer 1

这就是你想要的:

In [50]:
df.groupby('Col_1')[['Col_2','Col_3']].apply(lambda x: [tuple(x) for x in x.values]).to_dict()

Out[50]:
{'Key1': [('ACURA', 'CL')],
 'Key2': [('ACURA', 'EL'), ('ACURA', 'ILX')],
 'Key3': [('ACURA', 'INTEGRA')],
 'Key4': [('ACURA', 'INTEGRA')]}

在这里，我们对“Col_1”groupby，然后将值转换为列表中的元组，并在结果上调用 to_dict