如何从CSV中创建一个包含两列的字典作为Python中的键

发布时间：2020-12-20 11:03:38 所属栏目：Python 来源：网络整理

导读：参见英文答案 python pandas dataframe to dictionary????????????????????????????????????9个我正在尝试从具有3列的CSV创建字典,其中前两列成为键,第三列成为值：在此示例中,example.csv包含： Column-1 Column-2 Column-31 A foo2 B bar 预期产量应为：

参见英文答案 > python pandas dataframe to dictionary????????????????????????????????????9个
我正在尝试从具有3列的CSV创建字典,其中前两列成为键,第三列成为值：

在此示例中,example.csv包含：

Column-1    Column-2   Column-3
1           A          foo
2           B          bar

预期产量应为：

dictionary = {1,A: foo,2,B: bar}

我目前正在尝试从pandas数据框导入并转换为字典.我使用以下不成功：

df = pd.read_csv("example.csv")
dictionary = df.set_index(['Column-1','Column-2']).to_dict()

有没有办法使用pandas来创建字典,还是有更优雅的方式将csv转换为字典？

解决方法

您已接近IIUC – 需要通过[‘Column-3’]选择列：

d = df.set_index(['Column-1','Column-2'])['Column-3'].to_dict()
print (d)
{(2,'B'): 'bar',(1,'A'): 'foo'}

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!