Pandas: как получить имя столбца по индексу

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

Способ 1: получить имя столбца по позиции индекса

 #get column name in index position 2
colname = df. columns [ 2 ]

Способ 2: получить несколько имен столбцов по позициям индекса

 #get column names in index positions 2 and 4
colname = df. columns [[ 2 , 4 ]] 

В следующих примерах показано, как использовать каждый метод на практике со следующим 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],
                   ' steals ': [4, 3, 3, 2, 5, 4, 3, 8],
                   ' blocks ': [1, 0, 0, 3, 2, 2, 1, 5]})

#view DataFrame
print (df)

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

Пример 1. Получение имени столбца по позиции индекса

Следующий код показывает, как получить имя столбца в индексной позиции 2 DataFrame:

 #get column name for column in index position 2
colname = df. columns [ 2 ]

#display column name
print (colname)

assists

Имя столбца в позиции индекса 2 поддерживается .

Примечание . Значения индекса столбца в Python начинаются с 0. Таким образом, столбец с индексной позицией 2 будет третьим столбцом DataFrame с именем Assets .

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

Следующий код показывает, как получить имена столбцов в индексных позициях 2 и 4 DataFrame:

 #get column names in index positions 2 and 4
colname = df. columns [[ 2 , 4 ]] 

#display column names
print (colname)

Index(['assists', 'steals'], dtype='object')

По результату мы видим:

• Имя столбца в позиции индекса 2 поддерживается .
• Имя столбца в позиции индекса 4 — Stees .

Примечание . В этом примере мы решили получать имена столбцов для двух столбцов по позиции индекса, но вы можете использовать аналогичный синтаксис, чтобы получить имена столбцов для любого количества столбцов.

Дополнительные ресурсы

В следующих руководствах объясняется, как выполнять другие распространенные задачи в pandas:

Как получить значение ячейки из Pandas DataFrame
Как переименовать индекс в Pandas DataFrame
Как сортировать столбцы по имени в Pandas