Pandas: як швидко перетворити стовпець на список

Ви можете використати один із наведених нижче методів, щоб перетворити стовпець pandas DataFrame на список:

Спосіб 1: Використовуйте 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() для перетворення стовпця «points» 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() для перетворення стовпця «points» 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() має тенденцію бути найшвидшим.

Додаткові ресурси

У наступних посібниках пояснюється, як виконувати інші поширені функції зі стовпцями pandas DataFrame:

Як видалити стовпці в Pandas
Як виключити стовпці в Pandas
Як застосувати функцію до вибраних стовпців у Pandas
Як змінити порядок стовпців у Pandas DataFrame