来自 pandas 数据框列的 dict

Posted 2023-03-11

技术标签:

【中文标题】来自 pandas 数据框列的 dict【英文标题】：dict from columns of pandas dataframe columns 【发布时间】：2017-05-02 17:49:52 【问题描述】：

如何从 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

【问题讨论】：

【参考方案1】：

您可以使用 set_index 和 Series.to_dict - MutiIndex 创建 tuples：

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

【讨论】：

【参考方案2】：

我是这样做的

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

【讨论】：

以上是关于来自 pandas 数据框列的 dict的主要内容，如果未能解决你的问题，请参考以下文章