Преобразование панды DataFrame в словарь, используя несколько столбцов для ключа

Question

У меня есть панд 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, чтобы создать dict от DataFrame.

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} 

Сначала мы явно захватить необходимые столбцы из вашего кадра данных в требуемом порядке. Затем перебирайте строки и для каждой строки, создайте кортеж из адресов электронной почты (первые два столбца) и используйте это как ключ. Значение представляет собой просто 3-й столбец (email_count)