Pandas: cara menambahkan kolom baru dengan nomor baris

Ada dua cara untuk menambahkan kolom baru yang berisi nomor baris ke DataFrame pandas:

Metode 1: Gunakan tugas()

 df = df. assign (row_number= range ( len (df)))

Metode 2: Gunakan reset_index()

 df[' row_number '] = df. reset_index (). index

Kedua metode menghasilkan hasil yang sama.

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

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H'],
                   ' points ': [5, 17, 7, 19, 12, 13, 9, 24],
                   ' assists ': [4, 7, 7, 6, 8, 7, 10, 11]})

#view DataFrame
print (df)

  team points assists
0 to 5 4
1 B 17 7
2 C 7 7
3 D 19 6
4 E 12 8
5 F 13 7
6 G 9 10
7:24 a.m. 11

Contoh 1: Gunakan tugas() untuk menambahkan kolom nomor baris

Kode berikut menunjukkan cara menggunakan fungsi tugas() untuk menambahkan kolom baru bernama row_number yang menampilkan nomor baris setiap baris di DataFrame:

 #add column that contains row numbers
df = df. assign (row_number= range ( len (df)))

#view updated DataFrame
print (df)

  team points assists row_number
0 to 5 4 0
1 B 17 7 1
2 C 7 7 2
3 D 19 6 3
4 E 12 8 4
5 F 13 7 5
6 G 9 10 6
7:24 11 7

Perhatikan bahwa nilai pada kolom row_number berkisar antara 0 hingga 7.

Contoh 2: Gunakan reset_index() untuk menambahkan kolom nomor baris

Kode berikut menunjukkan cara menggunakan fungsi reset_index() untuk menambahkan kolom baru bernama row_number yang menampilkan nomor baris setiap baris di DataFrame:

 #add column that contains row numbers
df[' row_number '] = df. reset_index (). index

#view updated DataFrame
print (df)

  team points assists row_number
0 to 5 4 0
1 B 17 7 1
2 C 7 7 2
3 D 19 6 3
4 E 12 8 4
5 F 13 7 5
6 G 9 10 6
7:24 11 7

Perhatikan bahwa nilai pada kolom row_number berkisar antara 0 hingga 7.

Ini cocok dengan hasil contoh sebelumnya.

Sumber daya tambahan

Tutorial berikut menjelaskan cara melakukan tugas umum lainnya di panda:

Pandas: cara mencari selisih antara dua kolom
Pandas: cara mencari perbedaan antara dua garis
Pandas: cara mengurangi dua kolom