Як перетворити рядок pandas dataframe на список (з прикладом)

Ви можете використовувати наступний базовий синтаксис, щоб перетворити рядок pandas DataFrame на список:

 row_list = df. loc [2 ,:]. values . flatten (). tolist ()

Цей конкретний синтаксис перетворює значення в позиції 2 індексу рядка DataFrame у список.

У наступному прикладі показано, як використовувати цей синтаксис на практиці.

Приклад: перетворити рядок Pandas DataFrame на список

Припустімо, що у нас є такий фрейм даних pandas, який містить інформацію про різних баскетболістів:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H'],
                   ' points ': [18, 22, 19, 14, 14, 11, 20, 28],
                   ' assists ': [5, 7, 7, 9, 12, 9, 9, 4],
                   ' rebounds ': [11, 8, 10, 6, 6, 5, 9, 12]})

#view DataFrame
print (df)

  team points assists rebounds
0 A 18 5 11
1 B 22 7 8
2 C 19 7 10
3 D 14 9 6
4 E 14 12 6
5 F 11 9 5
6 G 20 9 9
7:28 4 12

Ми можемо використовувати такий синтаксис, щоб перетворити значення з позиції 2 індексу рядка в список:

 #convert row at index 2 to list
row_list = df. loc [2,:]. values . flatten (). tolist ()

#view results
print (row_list)

['C', 19, 7, 10]

Ми бачимо, що значення в позиції 2 індексу рядка були перетворені в список із чотирьох значень.

Ми можемо підтвердити, що результат справді є списком, використовуючи функцію type() :

 #view type
print ( type (row_list))

<class 'list'>

Якщо ви хочете, щоб у список були включені лише певні значення стовпців, ви можете вказати стовпці за назвою.

Наприклад, ми можемо використовувати такий синтаксис, щоб перетворити значення позиції 2 індексу рядка в список лише для стовпців команди та балів :

 #convert values in row index position 2 to list (for team and points columns)
row_list = df. loc [2,[' team ',' points ']]. values . flatten (). tolist ()

#view results
print (row_list)

['C', 19]

Зауважте, що до списку включено лише значення стовпців команди та балів .

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

У наступних посібниках пояснюється, як виконувати інші типові завдання в pandas:

Як перетворити певні стовпці в Pandas на масив NumPy
Як перетворити список на стовпець у Pandas
Як додати підсумковий рядок до Pandas DataFrame