Pandas 将DataFrame转化为字典列表

在本文中，我们将介绍如何将Pandas DataFrame转化为列表字典，这是在数据分析和处理中非常常见的一个任务。

阅读更多：Pandas 教程

Pandas DataFrame简介

在开始介绍如何将DataFrame转化为字典列表之前，让我们先了解一下Pandas DataFrame。

DataFrame是一个类似于SQL表格的数据结构。每个DataFrame都有行索引和列索引。可以通过列名来访问DataFrame中的数据，并且可以对DataFrame进行多种多样的操作，如筛选、排序、分组等。Pandas是Python数据科学最常用的库之一，是数据分析、机器学习等领域的基础库之一。

下面是一个示例DataFrame：

import pandas as pd

df = pd.DataFrame({
    'name': ['John', 'Lisa', 'Robert'],
    'age': [25, 31, 19],
    'gender': ['male', 'female', 'male']
})

print(df)

输出结果：

     name  age  gender
0    John   25    male
1    Lisa   31  female
2  Robert   19    male

将DataFrame转化为字典列表

在很多情况下，我们需要将DataFrame转化为字典列表。例如，我们需要将转化后的数据传递给其他函数或者Web API。

我们可以通过以下代码将DataFrame转化为字典列表：

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

其中，orient参数指定了转化后的格式。'records'表示每个行转化为一个字典，整个DataFrame就是由多个字典组成的列表。

输出结果：

[{'name': 'John', 'age': 25, 'gender': 'male'},
 {'name': 'Lisa', 'age': 31, 'gender': 'female'},
 {'name': 'Robert', 'age': 19, 'gender': 'male'}]

如果我们想要将DataFrame的某一列作为字典的键，可以将orient设置为'dict'，并指定列名：

dict_ = df.set_index('name').to_dict(orient='dict')['age']
print(dict_)

输出结果：

{'John': 25, 'Lisa': 31, 'Robert': 19}

如果我们只需要转化DataFrame的某几列，并且需要指定每个字典的键名，可以使用zip函数：

dict_list = [dict(zip(['name', 'age'], row)) for row in df[['name', 'age']].values]
print(dict_list)

输出结果：

[{'name': 'John', 'age': 25},
 {'name': 'Lisa', 'age': 31},
 {'name': 'Robert', 'age': 19}]

总结

本文介绍了如何将Pandas DataFrame转化为字典列表。我们可以使用to_dict函数将整个DataFrame转化为字典列表，也可以使用zip函数并指定每个字典的键名来转化需要的列。这个技巧在数据处理和分析中非常实用。