Pandas：如何快速将列转换为列表

您可以使用以下方法之一将 pandas DataFrame 的列转换为列表：

方法一：使用tolist()

 df[' my_column ']. tolist ()

方法2：使用list()

 list(df[' my_column '])

两种方法都会返回完全相同的结果。

以下示例展示了如何将这些方法与以下 pandas DataFrame 一起使用：

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'A', 'A', 'B', 'B', 'B'],
                   ' points ': [99, 90, 93, 86, 88, 82],
                   ' assists ': [33, 28, 31, 39, 34, 30]})

#view DataFrame
print (df)

  team points assists
0 to 99 33
1 A 90 28
2 A 93 31
3 B 86 39
4 B 88 34
5 B 82 30

方法 1：使用 tolist() 将列转换为列表

以下代码展示了如何使用tolist()函数将 DataFrame 的“points”列转换为列表：

 #convert column to list
my_list = df[' points ']. tolist ()

#view list
print (my_list)

[99, 90, 93, 86, 88, 82]

我们可以使用type()函数确认结果是一个列表：

 #check datatype
type (my_list)

list

方法 2：使用 list() 将列转换为列表

以下代码展示了如何使用list()函数将 DataFrame 的“points”列转换为列表：

 #convert column to list
my_list = list(df[' points '])
#view list
print (my_list)

[99, 90, 93, 86, 88, 82]

我们可以使用type()函数确认结果是一个列表：

 #check data type
type (my_list)

list

请注意，这两种方法返回完全相同的结果。

请注意，对于非常大的 DataFrame， tolist()方法往往是最快的。

其他资源

以下教程解释了如何使用 pandas DataFrame 的列执行其他常见功能：

如何删除 Pandas 中的列
如何排除 Pandas 中的列
如何将函数应用于 Pandas 中的选定列
如何更改 Pandas DataFrame 中的列顺序