使用多列从dataframe创建字典

2024-04-28 05:42:29 发布

您现在位置:Python中文网/ 问答频道 /正文

我需要创建一个字典,其中每个元素都是一行。键表示一个特定列,值将是剩余列条目的列表。下面是一个简单的例子:

示例df:

Col_A     Col_B    Col_C    Col_D
Roger      5        blue    house
Kim        2        green   car
Mike       10       red     bike

所需词典={'Roger':['5'、'blue'、'house']、'Kim':['2'、'green'、'car']、'Mike':['10'、'red'、'bike']}

任何建议都将不胜感激


Tags: 目的元素列表字典colgreenbluered
2条回答

将索引设置为COl_A,然后转置帧并转换为方向设置为list的字典

df.set_index('Col_A').T.to_dict('list')

{'Roger': [5, 'blue', 'house'],
 'Kim': [2, 'green', 'car'],
 'Mike': [10, 'red', 'bike']}

尝试:

print(df.set_index("Col_A").apply(list, 1).to_dict())

印刷品:

{'Roger': [5, 'blue', 'house'], 'Kim': [2, 'green', 'car'], 'Mike': [10, 'red', 'bike']}

相关问题 更多 >