Pandas: cara mendapatkan indeks kolom dari nama kolom

Anda bisa menggunakan metode berikut untuk mendapatkan nilai indeks kolom dari nama kolom di pandas:

Metode 1: Dapatkan indeks kolom untuk nama kolom

 df. columns . get_loc (' this_column ')

Metode 2: Dapatkan indeks kolom untuk beberapa nama kolom

 cols = [' this_column ', ' that_column ']

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

Contoh berikut menunjukkan cara menggunakan setiap metode dalam praktik dengan pandas DataFrame berikut:

 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

Contoh 1: Dapatkan indeks kolom untuk nama kolom

Kode berikut menunjukkan cara mendapatkan nilai indeks kolom untuk kolom bernama “returns”:

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

2

Kolom dengan nama “return” memiliki nilai indeks kolom 2 .

Catatan : Nilai indeks kolom dimulai dari 0 dengan Python. Jadi, karena “return” adalah kolom ketiga dari DataFrame, nilai indeksnya adalah 2.

Contoh 2: Dapatkan indeks kolom untuk beberapa nama kolom

Kode berikut menunjukkan cara mendapatkan nilai indeks kolom untuk beberapa kolom di 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]

Dari hasilnya kita dapat melihat:

• Kolom dengan nama “store” memiliki nilai indeks kolom 0 .
• Kolom dengan nama “return” memiliki nilai indeks kolom 2 .
• Kolom bernama “pengingat” memiliki nilai indeks kolom 3 .

Catatan : Anda dapat menemukan dokumentasi lengkap fungsi pandas get_loc() di sini .

Sumber daya tambahan

Tutorial berikut menjelaskan cara melakukan tugas umum lainnya di panda:

Cara mendapatkan nilai sel dari Pandas DataFrame
Cara mengganti nama indeks di Pandas DataFrame
Cara mengurutkan kolom berdasarkan nama di Pandas