Як вибрати стовпці за назвою в pandas (3 приклади)

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

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

 df. loc [:, ' column1 ']

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

 df. loc [:,[' column1 ',' column3 ',' column4 ']]

Спосіб 3: Виберіть стовпці в діапазоні за назвою

 df. loc [:, ' column2 ':' column4 ']

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

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' mavs ': [10, 12, 14, 15, 19, 22, 27],
                   ' cavs ': [18, 22, 19, 14, 14, 11, 20],
                   ' hornets ': [5, 7, 7, 9, 12, 9, 14],
                   ' spurs ': [10, 12, 14, 13, 13, 19, 22],
                   ' net ': [10, 14, 25, 22, 25, 17, 12]})

#view DataFrame
print (df)

   mavs cavs hornets spurs nets
0 10 18 5 10 10
1 12 22 7 12 14
2 14 19 7 14 25
3 15 14 9 13 22
4 19 14 12 13 25
5 22 11 9 19 17
6 27 20 14 22 12

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

Наступний код показує, як вибрати стовпець «шпори» у DataFrame:

 #select column with name 'spurs'
df. loc [:, ' spurs ']

0 10
1 12
2 14
3 13
4 13
5 19
6 22
Name: spurs, dtype: int64

Повертаються лише значення в стовпці «шпори».

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

Наступний код показує, як вибрати стовпці cavs, spurs і net у DataFrame:

 #select columns with names cavs, spurs, and nets
df. loc [:, [' cavs ', ' spurs ', ' nets ']]

        cavs spurs nets
0 18 10 10
1 22 12 14
2 19 14 25
3 14 13 22
4 14 13 25
5 11 19 17
6 20 22 12

Повертаються лише значення cavs, шпорів і стовпців мережі.

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

Наступний код показує, як вибрати всі стовпці між назвами «hornets» і «net» у DataFrame:

 #select all columns between hornets and nets
df. loc [:, ' hornets ': ' nets ']

        hornets spurs nets
0 5 10 10
1 7 12 14
2 7 14 25
3 9 13 22
4 12 13 25
5 9 19 17
6 14 22 12

Повертаються всі стовпці між назвами “hornets” і “net”.

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

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

Pandas: Як перемістити стовпець перед DataFrame
Pandas: як перевірити, чи містить стовпець рядок
Pandas: Як додати порожній стовпець до DataFrame (3 приклади)