Pandas: cara memilih kolom yang berisi string tertentu

Anda dapat menggunakan metode berikut untuk memilih kolom yang berisi string tertentu di pandas DataFrame:

Metode 1: Pilih kolom yang berisi string tertentu

 df. filter (regex=' string1 ')

Metode 2: Pilih kolom yang berisi salah satu dari beberapa string

 df. filter (regex=' string1|string2|string3 ')

Contoh berikut menunjukkan cara menggunakan masing-masing metode ini dalam praktik dengan pandas DataFrame berikut:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' mavs ': [10, 12, 14, 15, 19, 22, 27],
                   ' cavs ': [18, 22, 19, 14, 14, 11, 20],
                   ' hornets ': [5, 7, 7, 9, 12, 9, 14],
                   ' spurs ': [10, 12, 14, 13, 13, 19, 22],
                   ' net ': [10, 14, 25, 22, 25, 17, 12]})

#view DataFrame
print (df)

   mavs cavs hornets spurs nets
0 10 18 5 10 10
1 12 22 7 12 14
2 14 19 7 14 25
3 15 14 9 13 22
4 19 14 12 13 25
5 22 11 9 19 17
6 27 20 14 22 12

Contoh 1: Pilih kolom yang berisi string tertentu

Kode berikut menunjukkan cara menggunakan fungsi filter() untuk memilih hanya kolom yang berisi string “avs” di suatu tempat di namanya:

 #select columns that contain 'avs' in the name
df2 = df. filter (regex=' avs ')

#view DataFrame
print (df2)

   mavs cavs
0 10 18
1 12 22
2 14 19
3 15 14
4 19 14
5 22 11
6 27 20

Hanya kolom dengan nama “avs” yang dikembalikan.

Dalam hal ini, “mavs” dan “cavs” adalah satu-satunya kolom yang dikembalikan.

Contoh 2: Pilih kolom yang berisi salah satu dari beberapa string

Kode berikut menunjukkan cara menggunakan fungsi filter() untuk memilih hanya kolom yang berisi “avs” atau “ets” di suatu tempat di namanya:

 #select columns that contain 'avs' in the name
df2 = df. filter (regex=' avs|ets ')

#view DataFrame
print (df2)

   mavs cavs hornets nets
0 10 18 5 10
1 12 22 7 14
2 14 19 7 25
3 15 14 9 22
4 19 14 12 25
5 22 11 9 17
6 27 20 14 12

Hanya kolom dengan nama “avs” atau “ets” yang dikembalikan.

Perhatikan bahwa bilah vertikal ( | ) adalah operator ” OR ” di panda.

Jangan ragu untuk merangkai sebanyak mungkin operator “ATAU” ini untuk memilih kolom yang berisi salah satu dari banyak string berbeda.

Sumber daya tambahan

Tutorial berikut menjelaskan cara melakukan tugas umum lainnya di panda:

Pandas: Cara memindahkan kolom di depan DataFrame
Pandas: cara memeriksa apakah kolom berisi string
Pandas: Cara menambahkan kolom kosong ke DataFrame (3 contoh)