python - 迭代 df 以在 python 中创建嵌套字典

Question

我有一个客户 df，其中包含 ID、电子邮件、job_role 等列。

我想创建一个具有以下结构的字典:

{id1 : { 'email':'email_address@gmail.com', 'job_role':'manager', ...}
 id2: { 'email':'email_address2@aol.com', 'job_role':'associate', ...}}

有没有一种简单/可迭代的方法来做到这一点？

Answer 1

这就是 to_dict 方法的用途，带有“index”参数:https://pandas.pydata.org/pandas-docs/stable/generated/pandas.DataFrame.to_dict.html

>>> df = pandas.DataFrame({'email':['email_address@gmail.com','email_address2@aol.com'],'job_role':['manager','associate']})
>>> df
                     email   job_role
0  email_address@gmail.com    manager
1   email_address2@aol.com  associate
>>> df.to_dict('index')
{0: {'email': 'email_address@gmail.com', 'job_role': 'manager'}, 1: {'email': 'email_address2@aol.com', 'job_role': 'associate'}}