Python从数据框中循环创建字典

Question

第一篇文章！

我正在尝试创建一个从数据框循环创建字典的函数。 假设这两个简单数据框已经存在：

data1 = {'A':[1, 2, 3, 4], 'B':[5, 6, 7, 8]}
df1 = pd.DataFrame(data) 

dataframe1

和

data2 = {'C':[9, 10], 'D':[11, 12], 'E':[13, 14] }
df2 = pd.DataFrame(data2)

dataframe2

我希望能够创建这样的函数：

def create_dict(df):

df1的最终结果是：

dict1 = { 'A' : 1, 'B' : 5}
dict2 = { 'A' : 2, 'B' : 6}
dict3 = { 'A' : 3, 'B' : 7}
dict4 = { 'A' : 4, 'B' : 8}

df2的最终结果是：

dict1 = { 'C' : 9, 'D' : 11, 'E' : 13}
dict2 = { 'C' : 10, 'D' : 12, 'E' : 14}

我一直在寻找字典理解来解决这个问题，但是我显然不确定如何处理这个问题。谢谢！

Answer 1

将pandas.DataFrame.to_dict与records一起使用：

df1.to_dict(orient="records")

输出：

[{'A': 1, 'B': 5}, {'A': 2, 'B': 6}, {'A': 3, 'B': 7}, {'A': 4, 'B': 8}]