深入探究pandas.dataframeto_dict()

发布时间：2023-05-19

在pandas工具中，dataframe是一个重要的数据结构，而dataframe.to_dict()则是一个常用的数据转换函数。它能够将dataframe类型的数据转换成Python字典，让数据在字典结构下更容易地读取和使用。本篇文章将通过多个方面，从基本使用、参数设置、数据类型转换、性能优化等方面对dataframe.to_dict()做详细的阐述。

一、基本使用

首先，我们来看一下dataframe.to_dict()函数的基本使用方法。

import pandas as pd
# 首先，定义一个dataframe
df = pd.DataFrame({'name': ['Alice', 'Bob', 'Charlie'],
                   'age': [25, 30, 35],
                   'sex': ['female', 'male', 'male']})
# 调用dataframe.to_dict()函数
dict_df = df.to_dict()
# 输出结果
print(dict_df)

运行结果如下：

{'name': {0: 'Alice', 1: 'Bob', 2: 'Charlie'},
 'age': {0: 25, 1: 30, 2: 35},
 'sex': {0: 'female', 1: 'male', 2: 'male'}}

可以看到，dataframe.to_dict()将dataframe按行进行转换，并将结果存储在一个嵌套字典里。字典的键表示列名，而值则是一个字典，其中键为行号，值为对应单元格的值。

二、参数设置

dataframe.to_dict()函数还拥有许多可选参数，可以根据需要进行设置。下面我们来介绍其中几个常用的参数：

a. orient

orient参数用于控制字典的方向，即行和列的方向。默认情况下，orient的值为'columns'，表示以列为基准进行转换。我们可以将orient的值设置为'index'，以行为基准进行转换。示例如下：

# orient为columns的情况
dict_df1 = df.to_dict(orient='columns')
print(dict_df1)
# orient为index的情况
dict_df2 = df.to_dict(orient='index')
print(dict_df2)

运行结果分别如下：

{'name': {0: 'Alice', 1: 'Bob', 2: 'Charlie'},
 'age': {0: 25, 1: 30, 2: 35},
 'sex': {0: 'female', 1: 'male', 2: 'male'}}
{0: {'name': 'Alice', 'age': 25, 'sex': 'female'},
 1: {'name': 'Bob', 'age': 30, 'sex': 'male'},
 2: {'name': 'Charlie', 'age': 35, 'sex': 'male'}}

可以看到，当orient的值为'columns'时，结果的键为列名，值为字典，键为行号，值为对应单元格的值；当orient的值为'index'时，结果的键为行号，值为字典，键为列名，值为对应单元格的值。

b. columns

columns参数用于选择需要转换的列名。例如，我们只需要将'age'和'sex'这两列转换成字典：

# 将'age'和'sex'这两列转换为字典
dict_df3 = df.to_dict(orient='records', columns=['age', 'sex'])
print(dict_df3)

运行结果如下：

[{'age': 25, 'sex': 'female'},
 {'age': 30, 'sex': 'male'},
 {'age': 35, 'sex': 'male'}]

可以看到，转换结果只包含'age'和'sex'这两列的数据。

三、数据类型转换

dataframe.to_dict()函数在数据类型转换方面也有一些值得注意的地方。下面我们来看一下几个例子。

a. 转换DataFrame对象时，将NaN值转换为None

如果dataframe中存在NaN值，转换成字典时可以选择将其转换成None值。在to_dict()方法中，可以将参数na_rep设置为None。示例如下：

# 定义一个dataframe，其中包含NaN值
df1 = pd.DataFrame({'A': [1, 2, None],
                    'B': [4, None, 6]})
# 转换成字典，将NaN值转换为None
dict_df4 = df1.to_dict(na_rep=None)
print(dict_df4)

运行结果如下：

{'A': {0: 1.0, 1: 2.0, 2: None}, 'B': {0: 4.0, 1: None, 2: 6.0}}

可以看到，NaN值已经被转换为了None值。

b. 将数据类型转换为Python基础数据类型

如果需要将转换后的数据类型转换为Python基础数据类型（如int、str、float等），可以在to_dict()方法中将参数int、str、float设置为True。例如，本例中我们可以将数据的值转换成整型：

# 定义一个包含整型、浮点型以及字符串类型的dataframe
df2 = pd.DataFrame({'A': [1, 2, 3],
                    'B': [4.0, 5.0, 6.0],
                    'C': ['a', 'b', 'c']})
# 将数据转换成整型
dict_df5 = df2.to_dict('list', int=True)
print(dict_df5)

运行结果如下：

{'A': [1, 2, 3], 'B': [4, 5, 6], 'C': ['a', 'b', 'c']}

可以看到，数据已经成功转换成了整型。

四、性能优化

在将大量数据转换成字典时，性能可能成为一个问题。在pandas中，还有一些技巧可以帮助我们提高性能。

a. 使用numpy的数据类型

在数据极大的情况下，使用numpy的数据类型会比Python的数据类型更高效。例如，数据的类型为float32，可以在转换时进行指定。

import numpy as np
# 定义一个包含float32类型数据的dataframe
df3 = pd.DataFrame({'A': np.array([1.0, 2.0, 3.0], dtype=np.float32),
                    'B': np.array([4.0, 5.0, 6.0], dtype=np.float32),
                    'C': np.array([7.0, 8.0, 9.0], dtype=np.float32)})
# 将数据转换成字典
dict_df6 = df3.to_dict('list', int=True)
print(dict_df6)

运行结果如下：

{'A': [1.0, 2.0, 3.0], 'B': [4.0, 5.0, 6.0], 'C': [7, 8, 9]}

可以看到，数据已经成功转换成了float32类型。

b. 使用DataFrame的values属性

当需要对dataframe的所有数据进行处理时，可以使用dataframe的values属性，将其转换为numpy ndarray类型，再使用to_dict()函数进行转换。

# 定义一个包含float32类型数据的dataframe
df4 = pd.DataFrame({'A': np.array([1.0, 2.0, 3.0], dtype=np.float32),
                    'B': np.array([4.0, 5.0, 6.0], dtype=np.float32),
                    'C': np.array([7.0, 8.0, 9.0], dtype=np.float32)})
# 将数据转换成numpy数组
arr = df4.values
# 将numpy数组转换成字典
dict_df7 = [{col: row[i] for i, col in enumerate(df4.columns)}
            for row in arr]
print(dict_df7)

运行结果如下：

[{'A': 1.0, 'B': 4.0, 'C': 7.0},
 {'A': 2.0, 'B': 5.0, 'C': 8.0},
 {'A': 3.0, 'B': 6.0, 'C': 9.0}]

可以看到，数据已经成功转换成了字典类型。

总结

到这里，关于dataframe.to_dict()函数的详细解释就结束了。我们通过从基本使用、参数设置、数据类型转换、性能优化几个方面，对该函数做了深入的探究。希望读者们在日常使用中能够更好地掌握该函数，提高开发效率。

推荐文章