Pandas: como obter o nome da coluna por índice

Você pode usar os seguintes métodos para obter o nome de uma coluna por posição de índice no pandas:

Método 1: obter o nome da coluna por posição do índice

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

Método 2: obter vários nomes de colunas por posições de índice

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

Os exemplos a seguir mostram como usar cada método na prática com o seguinte DataFrame do 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

Exemplo 1: Obtenha um nome de coluna por posição de índice

O código a seguir mostra como obter o nome da coluna na posição 2 do índice do DataFrame:

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

#display column name
print (colname)

assists

O nome da coluna na posição 2 do índice é assistido .

Nota : Os valores do índice da coluna começam em 0 em Python. Portanto, a coluna na posição 2 do índice seria a terceira coluna do DataFrame, chamada assists .

Exemplo 2: Obtenha vários nomes de colunas por posições de índice

O código a seguir mostra como obter os nomes das colunas nas posições de índice 2 e 4 do 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')

Pelo resultado podemos ver:

• O nome da coluna na posição 2 do índice é assistido .
• O nome da coluna na posição 4 do índice é Stees .

Nota : Neste exemplo, optamos por obter os nomes das colunas de duas colunas por posição do índice, mas você pode usar sintaxe semelhante para obter os nomes das colunas de quantas colunas desejar.

Recursos adicionais

Os tutoriais a seguir explicam como realizar outras tarefas comuns em pandas:

Como obter o valor da célula do Pandas DataFrame
Como renomear o índice no Pandas DataFrame
Como classificar colunas por nome no Pandas