如何从数据帧创建字典？

Question

dataframe如下所示：

ENV   INVOCATION SSM_ID ANA_ID   VALUE  
env1   invo1      A      oas      1.6  
env1  invo1      A      default  2.0  
env1  invo1      B      oas      0.8  
env1  invo2      C      oas      0.4  
env2  invo1      A      oas      3.1  
env2  invo2      B      default  0.6  

我想要一本像下面这样的字典

dic = {(env1,invo1,A,oas) : 1.6 , (env1,invo1,A,default) : 2.0 , (env1,invo1,B,oas) : 0.8 , (env1,invo2,C,oas) : 0.4 , (env2,invo1,A,oas) : 3.1 , (env2,invo2,B,default) : 0.6 }

Answer 1

只对除VALUE以外的所有列执行set_index：

print (df.set_index(list(df.columns[:-1]))["VALUE"].to_dict())

{('env1', 'invo1', 'A', 'oas'): 1.6,
 ('env1', 'invo1', 'A', 'default'): 2.0,
 ('env1', 'invo1', 'B', 'oas'): 0.8,
 ('env1', 'invo2', 'C', 'oas'): 0.4,
 ('env2', 'invo1', 'A', 'oas'): 3.1,
 ('env2', 'invo2', 'B', 'default'): 0.6}

Answer 2

运行：

dict = df.set_index(['ENV', 'INVOCATION', 'SSM_ID', 'ANA_ID']).VALUE.to_dict()