Pandas: como obter o índice da coluna a partir do nome da coluna

Você pode usar os seguintes métodos para obter o valor do índice da coluna de um nome de coluna no pandas:

Método 1: obter o índice da coluna para um nome de coluna

 df. columns . get_loc (' this_column ')

Método 2: obter índice de coluna para vários nomes de colunas

 cols = [' this_column ', ' that_column ']

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

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 ({' 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

Exemplo 1: Obtenha o índice da coluna para um nome de coluna

O código a seguir mostra como obter o valor do índice da coluna chamada “returns”:

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

2

A coluna com o nome “return” possui um valor de índice de coluna de 2 .

Nota : Os valores do índice da coluna começam em 0 em Python. Portanto, como “return” é a terceira coluna do DataFrame, seu valor de índice é 2.

Exemplo 2: Obtenha o índice da coluna para vários nomes de colunas

O código a seguir mostra como obter o valor do índice da coluna para múltiplas colunas no 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]

Pelo resultado podemos ver:

• A coluna com o nome “store” possui um valor de índice de coluna igual a 0 .
• A coluna com o nome “return” possui um valor de índice de coluna de 2 .
• A coluna denominada “lembretes” possui um valor de índice de coluna de 3 .

Nota : Você pode encontrar a documentação completa da função get_loc() do pandas aqui .

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