Python如何将DataFrame指定列转换为非嵌套格式的键值对字典
Pandas DataFrame 转换为指定键值对字典的方法
你可以先将Keys列设置为DataFrame的索引,提取Atom_Vals列后再调用to_dict()方法即可得到目标结构,示例代码如下:
res = df.set_index('Keys')['Atom_Vals'].to_dict()
原方法df.to_dict('dict')默认返回以列名作为外层键的嵌套字典,不符合需求。上述写法不会修改原有DataFrame,set_index默认返回新的DataFrame对象,转换后会直接生成以Keys列值为键、Atom_Vals列值为值的一维字典。
你也可以使用zip拼接两列后直接转字典,作为替代方案:
res = dict(zip(df['Keys'], df['Atom_Vals']))
内容的提问来源于stack exchange,提问作者palutuna
相关产品推荐
相关产品推荐

