【Python】Pandas将 DataFrame 转换为列表

2024-05-31 06:29| 来源: 网络整理| 查看: 265

在数据分析和处理过程中，经常会使用到 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 中的数据转换操作。

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

【本文地址】

公司简介

联系我们