Cara mempertahankan kolom tertentu di pandas (dengan contoh)

Anda dapat menggunakan metode berikut untuk mempertahankan hanya kolom tertentu di pandas DataFrame:

Metode 1: Tentukan kolom mana yang akan disimpan

 #only keep columns 'col1' and 'col2'
df[[' col1 ', ' col2 ']]

Metode 2: Tentukan kolom yang akan dihapus

 #drop columns 'col3' and 'col4'
df[df. columns [~df. columns . isin ([' col3 ',' col4 '])]]

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

 import pandas as pd

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

#view DataFrame
df

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

Metode 1: Tentukan kolom mana yang akan disimpan

Kode berikut menunjukkan cara mendefinisikan DataFrame baru yang hanya mempertahankan kolom “tim” dan “poin”:

 #create new DataFrame and only keep 'team' and 'points' columns
df2 = df[[' team ', ' points ']]

#view new DataFrame
df2

        team points
0 to 11
1 to 7
2 to 8
3 B 10
4 B 13
5 B 13

Perhatikan bahwa DataFrame yang dihasilkan hanya mempertahankan dua kolom yang kami tentukan.

Metode 2: Tentukan kolom yang akan dihapus

Kode berikut menunjukkan cara mendefinisikan DataFrame baru yang menghapus kolom “attends” dan “bounce” dari DataFrame asli:

 #create new DataFrame and that drops 'assists' and 'rebounds'
df2 = df[df. columns [~df. columns . isin ([' assists ', ' rebounds '])]]

#view new DataFrame
df2

        team points
0 to 11
1 to 7
2 to 8
3 B 10
4 B 13
5 B 13

Perhatikan bahwa DataFrame yang dihasilkan menghapus kolom “bantuan” dan “pantulan” dari DataFrame asli dan mempertahankan kolom yang tersisa.

Sumber daya tambahan

Tutorial berikut menjelaskan cara melakukan operasi umum lainnya di panda:

Cara menghapus kolom pertama di Pandas DataFrame
Cara menghapus kolom duplikat di Pandas
Cara menghapus kolom berdasarkan indeks di Pandas