Pandas: как быстро преобразовать столбец в список

Вы можете использовать один из следующих методов для преобразования столбца DataFrame pandas в список:

Способ 1: используйте tolist()

 df[' my_column ']. tolist ()

Способ 2: используйте список()

 list(df[' my_column '])

Оба метода вернут один и тот же результат.

В следующих примерах показано, как использовать каждый из этих методов со следующим DataFrame pandas:

 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 в список:

 #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 в список:

 #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

Обратите внимание, что оба метода возвращают одинаковые результаты.

Обратите внимание, что для очень больших DataFrames метод tolist() имеет тенденцию быть самым быстрым.

Дополнительные ресурсы

В следующих руководствах объясняется, как выполнять другие распространенные функции со столбцами DataFrame pandas:

Как удалить столбцы в Pandas
Как исключить столбцы в Pandas
Как применить функцию к выбранным столбцам в Pandas
Как изменить порядок столбцов в Pandas DataFrame