Pandy: jak uzyskać indeks kolumny z nazwy kolumny

Możesz użyć następujących metod, aby uzyskać wartość indeksu kolumny z nazwy kolumny w pandach:

Metoda 1: Pobierz indeks kolumny dla nazwy kolumny

 df. columns . get_loc (' this_column ')

Metoda 2: Uzyskaj indeks kolumny dla wielu nazw kolumn

 cols = [' this_column ', ' that_column ']

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

Poniższe przykłady pokazują, jak w praktyce używać każdej metody z następującą ramką DataFrame pand:

 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

Przykład 1: Pobierz indeks kolumny dla nazwy kolumny

Poniższy kod pokazuje, jak uzyskać wartość indeksu kolumny dla kolumny o nazwie „powroty”:

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

2

Kolumna o nazwie „return” ma indeks kolumny o wartości 2 .

Uwaga : Wartości indeksów kolumn zaczynają się od 0 w Pythonie. Ponieważ „powrót” jest trzecią kolumną ramki DataFrame, jej wartość indeksu wynosi 2.

Przykład 2: Pobierz indeks kolumny dla wielu nazw kolumn

Poniższy kod pokazuje, jak uzyskać wartość indeksu kolumny dla wielu kolumn w ramce 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]

Z wyniku możemy zobaczyć:

• Kolumna o nazwie „store” ma indeks kolumny o wartości 0 .
• Kolumna o nazwie „return” ma indeks kolumny o wartości 2 .
• Kolumna o nazwie „przypomnienia” ma wartość indeksu kolumny 3 .

Uwaga : pełną dokumentację funkcji pandas get_loc() można znaleźć tutaj .

Dodatkowe zasoby

Poniższe samouczki wyjaśniają, jak wykonywać inne typowe zadania w pandach:

Jak uzyskać wartość komórki z Pandas DataFrame
Jak zmienić nazwę indeksu w Pandas DataFrame
Jak sortować kolumny według nazwy w Pandach