Pandas: як використовувати loc для вибору кількох стовпців

Ви можете використовувати функцію loc у pandas, щоб вибрати кілька стовпців у DataFrame за міткою.

Ось найпоширеніші способи зробити це:

Спосіб 1: виберіть кілька стовпців за назвою

 df. loc [:,[' col2 ',' col4 ']]

Спосіб 2: виберіть усі стовпці в діапазоні

 df. loc [:, ' col2 ':' col4 ']

Наступні приклади показують, як використовувати кожен метод на практиці з такими pandas DataFrame:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'A', 'A', 'A', 'B', 'B', 'B', 'B'],
                   ' points ': [5, 7, 7, 9, 12, 9, 9, 4],
                   ' assists ': [11, 8, 10, 6, 6, 5, 9, 12],
                   ' rebounds ': [6, 7, 7, 6, 10, 12, 10, 9]})

#view DataFrame
print (df)

  team points assists rebounds
0 A 5 11 6
1 To 7 8 7
2 To 7 10 7
3 to 9 6 6
4 B 12 6 10
5 B 9 5 12
6 B 9 9 10
7 B 4 12 9

Приклад 1: виберіть кілька стовпців за назвою

У наступному коді показано, як використовувати функцію loc для вибору стовпців «points» і «bounces» DataFrame:

 #select points and rebounds columns
df. loc [:,[' points ',' rebounds ']]

        rebound points
0 5 6
1 7 7
2 7 7
3 9 6
4 12 10
5 9 12
6 9 10
7 4 9

Зауважте, що повертається кожен рядок у стовпцях «очки» та «підбирання».

Також зауважте, що порядок, у якому ви вказуєте стовпці у функції loc , є порядком, у якому вони повертатимуться.

Наприклад, ми можемо спочатку повернути стовпець «підбирання», а потім стовпець «очки»:

 #select rebounds and points columns
df. loc [:, [' rebounds ', ' points ']]

	rebound points
0 6 5
1 7 7
2 7 7
3 6 9
4 10 12
5 12 9
6 10 9
7 9 4

Приклад 2: виберіть усі стовпці в діапазоні

У наступному коді показано, як використовувати функцію loc для вибору всіх стовпців між стовпцями «points» і «bounces» у DataFrame:

 #select all columns between points and rebounds columns
df. loc [:, ' points ': ' rebounds ']

	points assists rebounds
0 5 11 6
1 7 8 7
2 7 10 7
3 9 6 6
4 12 6 10
5 9 5 12
6 9 9 10
7 4 12 9

Зверніть увагу, що повертаються всі стовпці між стовпцями «points» і «bounces» у DataFrame.

Примітка . Щоб вибрати стовпці за позицією індексу, скористайтеся функцією iloc .

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

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

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