python - 如何从具有两列的 CSV 创建字典作为 Python 中的键

<分区>

我正在尝试从具有 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'}

关于python - 如何从具有两列的 CSV 创建字典作为 Python 中的键，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/49686322/

相关文章：

python - 删除所有非字母字符并分成新列