Pandas 将数据框行转换为字典

在本文中，我们将介绍如何将Pandas数据框的行转换成字典。在数据处理中，经常需要使用字典来表示数据。Pandas提供了一个方便的方法来实现这一点。

阅读更多：Pandas 教程

使用to_dict()方法将行转换为字典

Pandas提供了to_dict()方法，可以将数据框或数据框的一行转换成字典。to_dict()方法有几个参数，最常用的是orient参数。orient参数指定将数据以什么格式输出，有四个选项：’dict’、’list’、’series’和’index’，其中’dict’表示输出字典格式。

我们可以使用如下代码将数据框的一行row转换成字典：

import pandas as pd

df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})
row = df.iloc[0]  # 取数据框的第一行
d = row.to_dict()
print(d)

输出：

{'A': 1, 'B': 4, 'C': 7}

我们还可以将整个数据框转换成字典，这样每一行都是一个字典：

import pandas as pd

df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})
d = df.to_dict(orient='records')
print(d)

输出：

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

将字典转换回行或数据框

我们也可以将字典转换回数据框或一行。这可以使用Pandas的DataFrame()、Series()和Index()函数。

我们可以使用如下代码将字典转换成数据框：

import pandas as pd

d = {'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]}
df = pd.DataFrame.from_dict(d)
print(df)

输出：

   A  B  C
0  1  4  7
1  2  5  8
2  3  6  9

我们可以使用如下代码将字典转换成一行：

import pandas as pd

d = {'A': 1, 'B': 4, 'C': 7}
row = pd.Series(d)
print(row)

输出：

A    1
B    4
C    7
dtype: int64

总结

Pandas的to_dict()方法和DataFrame()、Series()、Index()函数提供了方便的方法将数据框或字典之间进行转换。我们可以使用它们将数据处理得更加方便和高效。