python - 使用多列作为键将 Pandas DataFrame 转换为字典

Question

我有一个 pandas DataFrame 如下

From_email,To_email,email_count
110165.74@compuserve.com,klay@enron.com,1
2krayz@gte.net,klay@enron.com,1
"<""d@piassick"".@enron@enron.com>",klay@enron.com,1

我想将其更改为以下格式的字典

hrc_dict = {('110165.74@compuserve.com', 'klay@enron.com'): 1,
            ('2krayz@gte.net', 'klay@enron.com'): 1, 
            ('<"d@piassick".@enron@enron.com>', 'klay@enron.com '): 1}

最好的方法是什么？

Answer 1

您可以使用 dict comprehension从您的 DataFrame 创建 dict。

df = DataFrame({
    'From_email': ['110165.74@compuserve.com', '2krayz@gte.net', '<"d@piassick".@enron@enron.com>'],
    'To_email': ['klay@enron.com', 'klay@enron.com', 'klay@enron.com'],
    'email_count': [1, 1, 1]})

d = {tuple(x[:2]):x[2] for x in df[['From_email', 'To_email', 'email_count']].values}

首先，我们明确以所需的顺序从数据框中获取必要的列。然后迭代行并为每一行从电子邮件地址(前两列)创建一个元组并将其用作键。该值只是第三列 (email_count)