将 dataframe 转换为以键和字典列表为值的字典

Question

我希望将下面的 dataframe 转换为嵌套字典

df_2 = pd.DataFrame([{
  'id': "qt1",
  'email': 'tim@apple.com',
  'company_name': "11-50",
  'first_name': "aram",
  'last_name': "lord",
  'title': "ceo"},
 {'id': "qt2",
  'email': 'aaron@unke.com',
  'company_name': "unke3",
  'first_name': "mich",
  'last_name': "smith",
  'title': "vp"},
 { 'id': "qt3",
  'email': 'dar@tryiu.com',
  'company_name': "triop",
  'first_name': "sela",
  'last_name': "augus",
  'title': "cfo"}])

看起来像这样

{'enrichments': [
    {'id': 'qt1', 'email': 'tim@apple.com', 'company_name': 'Apple', 'first_name': 'aram', 'last_name': 'lord', 'title': 'ceo',}, 
    {'id': 'qt2','email': 'aaron@unke.com', 'company_name': 'unke3', 'first_name': 'mich', 'last_name': 'smith', 'title': 'vp',}, 
    {'id': 'qt3','email': 'dar@tryiu.com', 'company_name': 'triop', 'first_name': 'sela', 'last_name': 'augus', 'title': 'cfo',}
    ],}

我尝试使用 groupby 和 to_dict，但它不起作用。 有什么建议么？ 预先感谢您的帮助。

Answer 1

您可以使用不同方向的to_dict ：

new_dict = {'enrichments': df_2.to_dict(orient='records')}

给出：

{'enrichments': [{'id': 'qt1', 'email': 'tim@apple.com', 'company_name': '11-50', 'first_name': 'aram', 'last_name': 'lord', 'title': 'ceo'}, 
{'id': 'qt2', 'email': 'aaron@unke.com', 'company_name': 'unke3', 'first_name': 'mich', 'last_name': 'smith', 'title': 'vp'}, 
{'id': 'qt3', 'email': 'dar@tryiu.com', 'company_name': 'triop', 'first_name': 'sela', 'last_name': 'augus', 'title': 'cfo'}]}