Pandas: pilih baris yang dua kolomnya sama

Anda dapat menggunakan metode berikut untuk memilih baris dalam pandas DataFrame di mana dua kolomnya (atau tidak) sama:

Metode 1: Pilih baris yang dua kolomnya sama

 df. query (' column1 ==column2 ')

Metode 2: Pilih baris yang dua kolomnya tidak sama

 df. query (' column1 !=column2 ')

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

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' painting ': ['A', 'B', 'C', 'D', 'E', 'F'],
                   ' fail1 ': ['Good', 'Good', 'Bad', 'Bad', 'Good', 'Good'],
                   ' fail2 ': ['Good', 'Bad', 'Bad', 'Good', 'Good', 'Good']})

#view DataFrame
print (df)

  painting rater1 rater2
0 A Good Good
1 B Good Bad
2 C Bad Bad
3 D Bad Good
4 E Good Good
5 F Good Good

Contoh 1: Pilih baris yang dua kolomnya sama

Kita dapat menggunakan sintaks berikut untuk memilih hanya baris di DataFrame yang nilai kolom rater1 dan rater2 sama:

 #select rows where rater1 is equal to rater2
df. query (' rater1 ==rater2 ')

 painting rater1 rater2
0 A Good Good
2 C Bad Bad
4 E Good Good
5 F Good Good

Perhatikan bahwa hanya garis dimana miss1 dan miss2 sama yang dipilih.

Kita juga dapat menggunakan fungsi len() jika kita hanya ingin menghitung berapa banyak baris yang memiliki nilai sama di kolom rater1 dan rater2 :

 #count the number of rows where rater1 is equal to rater2
len ( df.query (' rater1 ==rater2 '))

4

Ini memberitahu kita bahwa ada 4 baris yang nilai kolom miss1 dan miss2 sama.

Contoh 2: Pilih baris yang dua kolomnya tidak sama

Kita dapat menggunakan sintaks berikut untuk memilih hanya baris-baris di DataFrame yang nilai kolom penilai1 dan penilai2 tidak sama:

 #select rows where rater1 is not equal to rater2
df. query (' rater1 !=rater2 ')

 painting rater1 rater2
1 B Good Bad
3 D Bad Good

Perhatikan bahwa hanya garis dimana miss1 dan miss2 tidak sama yang dipilih.

Sumber daya tambahan

Tutorial berikut menjelaskan cara melakukan tugas umum lainnya di panda:

Cara mengganti nama kolom di Pandas
Bagaimana cara menambahkan kolom ke Pandas DataFrame
Bagaimana mengubah urutan kolom di Pandas DataFrame