Pandas: cara menggunakan loc untuk memilih beberapa kolom

Anda dapat menggunakan fungsi loc di panda untuk memilih beberapa kolom di DataFrame berdasarkan label.

Berikut cara paling umum untuk melakukan ini:

Metode 1: Pilih beberapa kolom berdasarkan nama

 df. loc [:,[' col2 ',' col4 ']]

Metode 2: Pilih semua kolom dalam rentang tersebut

 df. loc [:, ' col2 ':' col4 ']

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

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'A', 'A', 'A', 'B', 'B', 'B', 'B'],
                   ' points ': [5, 7, 7, 9, 12, 9, 9, 4],
                   ' assists ': [11, 8, 10, 6, 6, 5, 9, 12],
                   ' rebounds ': [6, 7, 7, 6, 10, 12, 10, 9]})

#view DataFrame
print (df)

  team points assists rebounds
0 A 5 11 6
1 To 7 8 7
2 To 7 10 7
3 to 9 6 6
4 B 12 6 10
5 B 9 5 12
6 B 9 9 10
7 B 4 12 9

Contoh 1: pilih beberapa kolom berdasarkan nama

Kode berikut menunjukkan cara menggunakan fungsi loc untuk memilih kolom “titik” dan “pantulan” pada DataFrame:

 #select points and rebounds columns
df. loc [:,[' points ',' rebounds ']]

        rebound points
0 5 6
1 7 7
2 7 7
3 9 6
4 12 10
5 9 12
6 9 10
7 4 9

Perhatikan bahwa setiap baris di kolom “poin” dan “pantulan” dikembalikan.

Perhatikan juga bahwa urutan kolom yang Anda tentukan dalam fungsi loc adalah urutan pengembalian kolom tersebut.

Misalnya, kita dapat mengembalikan kolom “rebound” terlebih dahulu, lalu kolom “poin”:

 #select rebounds and points columns
df. loc [:, [' rebounds ', ' points ']]

	rebound points
0 6 5
1 7 7
2 7 7
3 6 9
4 10 12
5 12 9
6 10 9
7 9 4

Contoh 2: Pilih semua kolom dalam rentang tersebut

Kode berikut menunjukkan cara menggunakan fungsi loc untuk memilih semua kolom antara kolom “titik” dan “pentalan” di DataFrame:

 #select all columns between points and rebounds columns
df. loc [:, ' points ': ' rebounds ']

	points assists rebounds
0 5 11 6
1 7 8 7
2 7 10 7
3 9 6 6
4 12 6 10
5 9 5 12
6 9 9 10
7 4 12 9

Perhatikan bahwa semua kolom antara kolom “titik” dan “pentalan” di DataFrame dikembalikan.

Catatan : Untuk memilih kolom berdasarkan posisi indeks, gunakan fungsi iloc sebagai gantinya.

Sumber daya tambahan

Tutorial berikut menjelaskan cara melakukan operasi umum lainnya di panda:

Cara memilih baris berdasarkan beberapa kondisi menggunakan Pandas Loc
Cara memilih baris berdasarkan nilai kolom di Pandas
Cara memilih baris berdasarkan indeks di Pandas