如何在Python中将数据框转换为列表

在Python中可以使用多种库来处理数据框，例如pandas和numpy等库，这些库提供了丰富的函数和方法来转换数据框。本文将介绍在Python中如何将数据框转换为列表。

使用pandas库

pandas库提供了将数据框转换为列表的方法，可以使用values属性将数据框转换为二维数组，然后使用tolist()方法将二维数组转换为列表。

import pandas as pd

# 创建一个数据框
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})

# 将数据框转换为列表
lst = df.values.tolist()

print(lst)
# 输出：[[1, 4, 7], [2, 5, 8], [3, 6, 9]]

上述代码使用pd.DataFrame创建了一个数据框df，然后使用df.values.tolist()将数据框转换为列表。其中，values属性将数据框转换为二维数组，tolist()方法将二维数组转换为列表。

使用numpy库

numpy库也提供了将数据框转换为列表的方法，可以使用tolist()方法直接将数据框转换为列表。

import numpy as np
import pandas as pd

# 创建一个数据框
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})

# 将数据框转换为列表
lst = df.to_numpy().tolist()

print(lst)
# 输出：[[1, 4, 7], [2, 5, 8], [3, 6, 9]]

上述代码使用pd.DataFrame创建了一个数据框df，然后使用df.to_numpy().tolist()将数据框转换为列表。其中，to_numpy()方法将数据框转换为numpy数组，tolist()方法将numpy数组转换为列表。

使用列表推导式

除了使用pandas和numpy库提供的方法，还可以使用列表推导式将数据框转换为列表。

import pandas as pd

# 创建一个数据框
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})

# 将数据框转换为列表
lst = [[df.at[i, j] for j in df.columns] for i in df.index]

print(lst)
# 输出：[[1, 4, 7], [2, 5, 8], [3, 6, 9]]

上述代码使用pd.DataFrame创建了一个数据框df，然后使用列表推导式将数据框转换为列表。其中，使用df.at[i, j]获取数据框中第i行第j列的元素，通过列表推导式遍历数据框的每个元素，将每行转换为一个列表。

总结

本文介绍了在Python中将数据框转换为列表的三种方法，分别使用pandas、numpy库和列表推导式。使用这些方法可以很方便地将数据框转换为列表，方便后续的数据处理和分析。