Cara mendapatkan nomor baris di pandas dataframe

Seringkali Anda mungkin ingin mendapatkan nomor baris dalam pandas DataFrame yang berisi nilai tertentu.

Untungnya, hal ini mudah dilakukan dengan menggunakan fungsi .index .

Tutorial ini menunjukkan beberapa contoh penggunaan praktis fungsi ini.

Contoh 1: Dapatkan nomor baris yang sesuai dengan nilai tertentu

Misalkan kita memiliki panda DataFrame berikut:

 import pandas as pd

#createDataFrame
df = pd.DataFrame({'points': [25, 12, 15, 14, 19],
                   'assists': [5, 7, 7, 9, 12],
                   'team': ['Mavs', 'Mavs', 'Spurs', 'Celtics', 'Warriors']})

#view DataFrame 
print (df)

        team assists points
0 25 5 Mavs
1 12 7 Mavs
2 15 7 Spurs
3 14 9 Celtics
4 19 12 Warriors

Kita dapat menggunakan sintaks berikut untuk mendapatkan nomor baris di mana “tim” sama dengan Mavs:

 #get row numbers where 'team' is equal to Mavs
df[df[' team '] == ' Mavs ']. index

Int64Index([0, 1], dtype='int64')

Kita dapat melihat bahwa nama tim sama dengan ‘Mavs’ pada indeks baris 0 dan 1 .

Kita juga bisa mendapatkan nomor baris dimana nama tim ditemukan dalam daftar nama tim tertentu:

 #get row numbers where 'team' is equal to Mavs or Spurs
filter_list = [' Mavs ', ' Spurs ']

#return only rows where team is in the list of team names
df[df. team . isin (filter_list)]. index

Int64Index([0, 1, 2], dtype='int64')

Kita dapat melihat bahwa nama tim sama dengan ‘Mavs’ atau ‘Spurs’ pada indeks baris 0 , 1 dan 2 .

Contoh 2: Dapatkan nomor baris unik

Misalkan kita memiliki panda DataFrame berikut:

 import pandas as pd

#createDataFrame
df = pd.DataFrame({'points': [25, 12, 15, 14, 19],
                   'assists': [5, 7, 7, 9, 12],
                   'team': ['Mavs', 'Mavs', 'Spurs', 'Celtics', 'Warriors']})

Jika Anda mengetahui bahwa satu baris cocok dengan nilai tertentu, Anda dapat mengambil nomor baris unik tersebut menggunakan sintaks berikut:

 #get the row number where team is equal to Celtics
df[df[' team '] == ' Celtics ']. index [ 0 ]

3

Kita dapat melihat bahwa tim tersebut setara dengan “Celtics” pada indeks baris nomor 3 .

Contoh 3: Dapatkan jumlah nomor baris

Misalkan kita memiliki panda DataFrame berikut:

 import pandas as pd

#createDataFrame
df = pd.DataFrame({'points': [25, 12, 15, 14, 19],
                   'assists': [5, 7, 7, 9, 12],
                   'team': ['Mavs', 'Mavs', 'Spurs', 'Celtics', 'Warriors']})

Jika Anda ingin mengetahui jumlah total baris yang kolomnya sama dengan nilai tertentu, Anda dapat menggunakan sintaks berikut:

 #find total number of rows where team is equal to Mavs
len (df[df[' team '] == ' Celtics ']. index )

2

Kita dapat melihat bahwa tim ini setara dengan “Mavs” di total 2 baris.

Sumber daya tambahan

Tutorial berikut menjelaskan cara melakukan tugas umum lainnya di panda:

Cara menemukan nilai unik di beberapa kolom di Pandas
Cara memfilter Pandas DataFrame pada berbagai kondisi
Cara menghitung nilai yang hilang di Pandas DataFrame