pandas 转为list|极客教程

pandas 转为list

在数据处理和分析的工作中，经常会用到 pandas 这个强大的 Python 库。pandas 提供了 DataFrame 数据结构，能够方便地对数据进行操作和分析。有时候我们需要将 pandas 的 DataFrame 转换为 list 类型，这样方便进行数据的其他类型操作。本文将详细介绍如何将 pandas 转为 list，并给出多个示例代码。

方法一：使用 to_list() 方法

pandas 的 Series 和 DataFrame 对象都有一个 to_list() 方法，可以将其转换为 list 类型。下面是一个示例代码：

import pandas as pd

data = {'name': ['Alice', 'Bob', 'Cathy', 'David'],
        'age': [25, 26, 27, 28]}

df = pd.DataFrame(data)
name_list = df['name'].to_list()
print(name_list)

运行结果如下：

['Alice', 'Bob', 'Cathy', 'David']

上面的代码先创建一个包含两列数据的 DataFrame，然后使用 to_list() 方法将 ‘name’ 这一列转为 list 类型。

方法二：使用 values 属性

除了使用 to_list() 方法之外，还可以使用 DataFrame 的 values 属性将其转为一个包含所有数据的二维数组，然后再将其转为 list。以下是一个示例代码：

import pandas as pd

data = {'name': ['Alice', 'Bob', 'Cathy', 'David'],
        'age': [25, 26, 27, 28]}

df = pd.DataFrame(data)
data_list = df.values.tolist()
print(data_list)

运行结果如下：

[['Alice', 25], ['Bob', 26], ['Cathy', 27], ['David', 28]]

示例代码中，使用 values 属性获取 DataFrame 中的所有数据，然后调用 tolist() 方法将其转为 list 类型。

方法三：遍历转换

如果需要将 DataFrame 中的多列数据转为 list，可以遍历 DataFrame，并逐一将其转为 list。以下是一个示例代码：

import pandas as pd

data = {'name': ['Alice', 'Bob', 'Cathy', 'David'],
        'age': [25, 26, 27, 28],
        'gender': ['F', 'M', 'F', 'M']}

df = pd.DataFrame(data)
column_list = []

for col in df.columns:
    column_list.append(df[col].to_list())

print(column_list)

运行结果如下：

[['Alice', 'Bob', 'Cathy', 'David'],
 [25, 26, 27, 28],
 ['F', 'M', 'F', 'M']]

示例代码中，遍历 DataFrame 中的每一列数据，逐一将其转为 list，并添加到 column_list 中。

总结起来，本文介绍了三种将 pandas 转为 list 的方法：使用 to_list() 方法、使用 values 属性和遍历转换。这些方法能够方便地将 DataFrame 中的数据转为 list，方便进行进一步的数据处理和分析。

pandas 转为list