Panda: come ottenere l'indice della colonna dal nome della colonna

È possibile utilizzare i seguenti metodi per ottenere il valore dell’indice di colonna dal nome di una colonna in panda:

Metodo 1: ottenere l’indice di colonna per un nome di colonna

 df. columns . get_loc (' this_column ')

Metodo 2: ottieni l’indice di colonna per più nomi di colonna

 cols = [' this_column ', ' that_column ']

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

I seguenti esempi mostrano come utilizzare ciascun metodo nella pratica con i seguenti DataFrame panda:

 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

Esempio 1: ottenere l’indice di colonna per un nome di colonna

Il codice seguente mostra come ottenere il valore dell’indice di colonna per la colonna denominata “restituisce”:

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

2

La colonna con il nome “return” ha un valore di indice di colonna pari a 2 .

Nota : i valori dell’indice delle colonne iniziano da 0 in Python. Quindi, poiché “return” è la terza colonna del DataFrame, il suo valore di indice è 2.

Esempio 2: ottenere l’indice di colonna per più nomi di colonna

Il codice seguente mostra come ottenere il valore dell’indice di colonna per più colonne nel 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]

Dal risultato possiamo vedere:

• La colonna con il nome “store” ha un valore di indice di colonna pari a 0 .
• La colonna con il nome “return” ha un valore di indice di colonna pari a 2 .
• La colonna denominata “promemoria” ha un valore di indice di colonna pari a 3 .

Nota : puoi trovare la documentazione completa della funzione get_loc() di panda qui .

Risorse addizionali

I seguenti tutorial spiegano come eseguire altre attività comuni nei panda:

Come ottenere il valore della cella da Pandas DataFrame
Come rinominare l’indice in Pandas DataFrame
Come ordinare le colonne per nome in Pandas