Python Pandas将 DataFrame 转换为列表

更多Python学习内容：ipengtao.com

在数据分析和处理过程中，经常会使用到 Pandas 库来处理和操作数据。Pandas 提供了灵活强大的数据结构 DataFrame，它可以存储和处理各种类型的数据，并提供了丰富的方法和函数来进行数据操作。有时候，我们需要将 DataFrame 转换为列表形式，以便进一步处理或者与其他库进行集成。本文将详细介绍如何使用 Pandas 将 DataFrame 转换为列表，并提供丰富的示例代码来帮助读者更好地理解和应用。

DataFrame 数据结构简介

DataFrame 是 Pandas 中最常用的数据结构之一，它类似于电子表格或 SQL 表，可以存储二维数据，并且可以轻松处理不同类型的数据。DataFrame 由行和列组成，每一列可以是不同的数据类型（整数、浮点数、字符串等）。

Pandas 将 DataFrame 转换为列表的方法

Pandas 提供了多种方法来将 DataFrame 转换为列表，常用的方法包括使用 values 属性、to_numpy() 方法和列表解析等。接下来将分别介绍这些方法的用法和示例。

1 使用 `values` 属性

values 属性可以将 DataFrame 转换为 NumPy 数组，然后再将 NumPy 数组转换为列表。

import pandas as pd# 创建 DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})# 使用 values 属性将 DataFrame 转换为 NumPy 数组，然后再转换为列表
list_from_values = df.values.tolist()print("列表 from values 属性:", list_from_values)

2 使用 `to_numpy()` 方法

to_numpy() 方法可以将 DataFrame 直接转换为 NumPy 数组，然后再将 NumPy 数组转换为列表。

import pandas as pd# 创建 DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})# 使用 to_numpy() 方法将 DataFrame 转换为 NumPy 数组，然后再转换为列表
list_from_to_numpy = df.to_numpy().tolist()print("列表 from to_numpy() 方法:", list_from_to_numpy)

3 使用列表解析

列表解析是一种简洁高效的方式，可以将 DataFrame 中的每一行数据转换为列表。

import pandas as pd# 创建 DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})# 使用列表解析将 DataFrame 中的每一行数据转换为列表
list_from_list_comprehension = [list(row) for row in df.values]print("列表 from 列表解析:", list_from_list_comprehension)

示例代码演示

通过一个更具体的示例来演示如何将 DataFrame 转换为列表，并且对比不同方法的效果。

import pandas as pd# 创建 DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})# 使用 values 属性将 DataFrame 转换为列表
list_from_values = df.values.tolist()# 使用 to_numpy() 方法将 DataFrame 转换为列表
list_from_to_numpy = df.to_numpy().tolist()# 使用列表解析将 DataFrame 中的每一行数据转换为列表
list_from_list_comprehension = [list(row) for row in df.values]print("列表 from values 属性:", list_from_values)
print("列表 from to_numpy() 方法:", list_from_to_numpy)
print("列表 from 列表解析:", list_from_list_comprehension)

输出结果：

列表 from values 属性: [[1, 4], [2, 5], [3, 6]]
列表 from to_numpy() 方法: [[1, 4], [2, 5], [3, 6]]
列表 from 列表解析: [[1, 4], [2, 5], [3, 6]]

应用场景详解

将 DataFrame 转换为列表在实际应用中有着广泛的应用场景，下面将更详细地探讨一些常见的应用场景，并提供丰富的示例代码。

1 数据传递与集成

在数据处理和分析过程中，常常需要将 Pandas 的 DataFrame 数据传递给其他库或模块进行进一步的处理或分析。有些库可能更适合处理列表形式的数据，因此将 DataFrame 转换为列表可以方便地与这些库集成。

import pandas as pd
import numpy as np# 创建 DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})# 将 DataFrame 转换为列表
data_list = df.values.tolist()# 传递数据给其他库进行处理
processed_data = my_library.process(data_list)

2 API 调用

有些 API 可能要求输入数据的格式为列表形式，因此将 DataFrame 转换为列表可以方便地调用这些 API。

import pandas as pd
import requests# 创建 DataFrame
df = pd.read_csv('data.csv')# 将 DataFrame 转换为列表
data_list = df.values.tolist()# 调用 API
response = requests.post('https://api.example.com/data', json={'data': data_list})

3 数据导出

有时候，需要将 DataFrame 中的数据导出为文本文件或其他格式，而某些导出方式可能更适合处理列表形式的数据。

import pandas as pd# 创建 DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})# 将 DataFrame 转换为列表
data_list = df.values.tolist()# 导出为 CSV 文件
with open('data.csv', 'w') as f:for row in data_list:f.write(','.join(map(str, row)) + '\n')

4 数据展示与可视化

有些数据可视化库可能更适合处理列表形式的数据，因此将 DataFrame 转换为列表可以方便地展示数据或生成可视化图表。

import pandas as pd
import matplotlib.pyplot as plt# 创建 DataFrame
df = pd.read_csv('data.csv')# 将 DataFrame 转换为列表
data_list = df.values.tolist()# 绘制折线图
plt.plot(data_list)
plt.show()

总结

本文介绍了如何使用 Pandas 将 DataFrame 转换为列表，并提供了多种方法的示例代码。了解如何将 DataFrame 转换为列表可以帮助我们更灵活地处理和分析数据，并且可以与其他库或模块进行无缝集成。在实际应用中，根据具体需求选择合适的方法来进行数据转换，可以提高编程效率和代码质量。希望本文能够帮助大家更好地理解和应用 Pandas 中的数据转换操作。

如果你觉得文章还不错，请大家点赞、分享、留言下，因为这将是我持续输出更多优质文章的最强动力！

如果想要系统学习Python、Python问题咨询，或者考虑做一些工作以外的副业，都可以扫描二维码添加微信，围观朋友圈一起交流学习。