熊貓數據框列的字典

Question

如何從pandas數據框的列創建字典。 這是測試代碼：

import pandas as pd
df = pd.DataFrame({
    'c0': ['A','A','B'],
    'c1': ['b','c','d'],
    'c2': [1, 3,4],
    'c3': [5,7,8]})

輸出：

  c0 c1  c2  c3
0  A  b   1   5
1  A  c   3   7
2  B  d   4   8

我想從例如key：（c0，c1）和value：（c2）獲取字典

{('A', 'b'): 1, ('A', 'c'): 3, ('B', 'd'): 4}

Answer 1

可以將set_index與Series.to_dict MutiIndex使用MutiIndex創建tuples ：

print (df.set_index(['c0','c1'])['c2'].to_dict())
{('B', 'd'): 4, ('A', 'b'): 1, ('A', 'c'): 3}

Answer 2

這是我做的

dict(zip(zip(df['c0'], df['c1']), df['c2']))