Pandas: як отримати індекс стовпця з назви стовпця

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

Спосіб 1. Отримайте індекс стовпця для імені стовпця

 df. columns . get_loc (' this_column ')

Спосіб 2. Отримайте індекс стовпця для кількох імен стовпців

 cols = [' this_column ', ' that_column ']

[df. columns . get_loc (c) for c in cols if c in df]

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

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' store ': ['A', 'A', 'A', 'A', 'B', 'B', 'B', 'B'],
                   ' sales ': [18, 10, 14, 13, 19, 24, 25, 29],
                   ' returns ': [1, 2, 2, 3, 2, 3, 5, 4],
                   ' recalls ': [0, 0, 2, 1, 1, 2, 0, 1]})
#view DataFrame
print (df)

  store sales returns recalls
0 to 18 1 0
1 to 10 2 0
2 A 14 2 2
3 A 13 3 1
4 B 19 2 1
5 B 24 3 2
6 B 25 5 0
7 B 29 4 1

Приклад 1: Отримання індексу стовпця для імені стовпця

У наведеному нижче коді показано, як отримати значення індексу стовпця для стовпця з назвою “повернення”:

 #get column index for column with the name 'returns'
df. columns . get_loc (' returns ')

2

Стовпець із назвою “return” має значення індексу стовпця 2 .

Примітка : значення індексу стовпця починаються з 0 у Python. Отже, оскільки «повернення» є третім стовпцем DataFrame, значення його індексу дорівнює 2.

Приклад 2: Отримання індексу стовпця для кількох імен стовпців

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

 #define list of columns to get index for
cols = [' store ', ' returns ', ' recalls ']

#get column index for each column in list
[df. columns . get_loc (c) for c in cols if c in df]

[0, 2, 3]

З результату ми бачимо:

• Стовпець із назвою «магазин» має значення індексу стовпця 0 .
• Стовпець із назвою “return” має значення індексу стовпця 2 .
• Стовпець із назвою “нагадування” має значення індексу стовпця 3 .

Примітка . Ви можете знайти повну документацію функції pandas get_loc() тут .

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

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

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