将数据帧转换为字典

Question

我打算从数据框中获取一个列名作为键的字典.

假设我有一个数据帧:

    a    b
 0  ac   dc
 1  ddd  fdf

我希望输出为:

{a : ac, b : dc}

我希望它一行一行地完成.任何形式的帮助将不胜感激.在这里,我希望列名称作为结果字典中的键.

Answer 1

您可以使用该to_dict()方法orient='records'

import pandas as pd

df = pd.DataFrame([{'a': 'ac', 'b': 'dc'}, {'a': 'ddd', 'b': 'fdf'}])
print(df)

#      a    b
# 0   ac   dc
# 1  ddd  fdf

d = df.to_dict(orient='records')
print(d)

# [{'b': 'dc', 'a': 'ac'}, {'b': 'fdf', 'a': 'ddd'}]