如何在大熊猫中将行和列作为键将数据帧转换为字典?

时间:2019-07-03 10:20:18

标签: python pandas dataframe dictionary

我有一个由以下数据和功能构成的数据框


model.Crops = ["barley", "rapeseed", "wheat"]
model.FixedInputs = ["land", "labor", "capital"]
Beta =  [[0.3, 0.1, 0.3],\
        [0.2, 0.1, 0.2],\
        [0.3, 0.1, 0.2]]

pd.DataFrame(data=Beta_F_Data, index=model.Crops, columns=model.FixedInputs)

我得到了这个矩阵:

FixedInputs  land  labor  capital
Crops                            
barley        0.3    0.1      0.3
rapeseed      0.2    0.1      0.2
wheat         0.3    0.1      0.2

如何将该矩阵转换为以索引和列为键的字典?

我尝试了df.to_dict(),但它仅使用列作为键。

它应该看起来像这样:

dict = {(barley, land): 0.3, (barley, labor): 0.1, ...(wheat, capital):0.2}

1 个答案:

答案 0 :(得分:9)

在致电stack之前,您需要to_dict

df.stack().to_dict()

Out[389]:
{('barley', 'land'): 0.3,
 ('barley', 'labor'): 0.1,
 ('barley', 'capital'): 0.3,
 ('rapeseed', 'land'): 0.2,
 ('rapeseed', 'labor'): 0.1,
 ('rapeseed', 'capital'): 0.2,
 ('wheat', 'land'): 0.3,
 ('wheat', 'labor'): 0.1,
 ('wheat', 'capital'): 0.2}