Python | 熊猫 DataFrame.to_dict()

在 Python 中，熊猫（Pandas）是广泛使用的数据分析库之一。DataFrame.to_dict() 是熊猫中的一个函数，它用于将 DataFrame 转换为字典。本文将介绍 DataFrame.to_dict() 的用法和返回的字典格式。

用法

DataFrame 是熊猫中用于存储和操作数据的一个重要数据结构。当需要将 DataFrame 转换为字典时，可以使用 to_dict() 函数。

df.to_dict(orient='dict', into=dict)

to_dict() 函数接受两个可选参数：

• orient：指定返回的字典格式，可选值有 'dict'、'list'、'series'、'split' 和 'records'。
• into：指定返回的字典类型，默认为 dict。

字典格式

不同的 orient 可以生成不同形式的字典：

• 'dict'：默认选项，返回字典，以列名为键，以列数据为值的字典。形式为 {column_name: {index_value: value}}。
• 'list'：返回字典，以列名为键，以列数据的列表为值的字典。形式为 {column_name: [value1, value2, ...]}。
• 'series'：返回字典，以列名为键，以列数据的 Series 对象为值的字典。形式为 {column_name: Series_object}。
• 'split'：返回字典，每列数据的字典键以行索引和列名为元组：{index: {column_name: value}}。
• 'records'：返回字典，以行索引为键，以每一行数据为值的字典。形式为 {index: {column_name: value}}。

以下是一个示例：

import pandas as pd

data = {'name': ['Alice', 'Bob', 'Charlie', 'David'],
        'age': [25, 30, 35, 40],
        'country': ['USA', 'Canada', 'UK', 'Australia']}

df = pd.DataFrame(data)

# 将 DataFrame 转换为字典（默认格式）
dict_default = df.to_dict()
print("默认格式：", dict_default)

# 将 DataFrame 转换为字典（指定 orient='list'）
dict_list = df.to_dict(orient='list')
print("以列表为值的字典格式：", dict_list)

# 将 DataFrame 转换为字典（指定 orient='series'）
dict_series = df.to_dict(orient='series')
print("以 Series 为值的字典格式：", dict_series)

输出结果：

默认格式： {'name': {0: 'Alice', 1: 'Bob', 2: 'Charlie', 3: 'David'}, 'age': {0: 25, 1: 30, 2: 35, 3: 40}, 'country': {0: 'USA', 1: 'Canada', 2: 'UK', 3: 'Australia'}}
以列表为值的字典格式： {'name': ['Alice', 'Bob', 'Charlie', 'David'], 'age': [25, 30, 35, 40], 'country': ['USA', 'Canada', 'UK', 'Australia']}
以 Series 为值的字典格式： {'name': 0       Alice
                              1         Bob
                              2     Charlie
                              3       David
                              Name: name, dtype: object,
                           'age': 0    25
                                   1    30
                                   2    35
                                   3    40
                                   Name: age, dtype: int64,
                           'country': 0           USA
                                       1        Canada
                                       2            UK
                                       3    Australia
                                       Name: country, dtype: object}

总结

DataFrame.to_dict() 函数可以将熊猫的 DataFrame 转换为不同形式的字典，方便在数据处理和分析过程中进行字典操作。根据不同的需求，可以选择合适的 orient 参数来获取所需的字典格式。

注意：由于 DataFrame 可以包含混合类型的数据，转换为字典时可能会导致类型丢失或数据结构不一致的问题，请根据实际情况使用适当的字典格式。