Cara menggunakan metode assign() di pandas (dengan contoh)

Metode penetapan() dapat digunakan untuk menambahkan kolom baru ke DataFrame pandas.

Metode ini menggunakan sintaks dasar berikut:

 df. assign (new_column = values)

Penting untuk dicatat bahwa metode ini hanya akan menampilkan DataFrame baru di konsol, tetapi tidak akan benar-benar mengubah DataFrame asli.

Untuk mengubah DataFrame asli, Anda perlu menyimpan hasil metode penetapan() dalam variabel baru.

Contoh berikut menunjukkan cara menggunakan metode penetapan() dengan cara yang berbeda dengan pandas DataFrame berikut:

 import pandas as pd

#define DataFrame
df = pd. DataFrame ({' points ': [25, 12, 15, 14, 19, 23, 25, 29],
                   ' assists ': [5, 7, 7, 9, 12, 9, 9, 4],
                   ' rebounds ': [11, 8, 10, 6, 6, 5, 9, 12]})

#view DataFrame
print (df)

   points assists rebounds
0 25 5 11
1 12 7 8
2 15 7 10
3 14 9 6
4 19 12 6
5 23 9 5
6 25 9 9
7 29 4 12

Contoh 1: Tetapkan variabel baru ke DataFrame

Kode berikut menunjukkan cara menggunakan metode penetapan() untuk menambahkan variabel baru ke DataFrame bernama points2 yang nilainya sama dengan nilai pada kolom points dikalikan dua:

 #add new variable called points2
df. assign (points2 = df.points * 2 )

    points assists rebounds points2
0 25 5 11 50
1 12 7 8 24
2 15 7 10 30
3 14 9 6 28
4 19 12 6 38
5 23 9 5 46
6 25 9 9 50
7 29 4 12 58

Perhatikan bahwa metode penetapan() ini tidak mengubah DataFrame asli.

Jika kita mencetak DataFrame asli, kita akan melihat bahwa DataFrame tetap tidak berubah:

 #print original DataFrame
print (df)

   points assists rebounds
0 25 5 11
1 12 7 8
2 15 7 10
3 14 9 6
4 19 12 6
5 23 9 5
6 25 9 9
7 29 4 12

Untuk menyimpan hasil dari metode penetapan() , kita dapat menyimpan hasilnya di DataFrame baru:

 #add new variable called points2 and save results in new DataFrame
df. assign (points2 = df.points * 2 )

#view new DataFrame
print (df_new)

   points assists rebounds points2
0 25 5 11 50
1 12 7 8 24
2 15 7 10 30
3 14 9 6 28
4 19 12 6 38
5 23 9 5 46
6 25 9 9 50
7 29 4 12 58

DataFrame baru bernama df_new sekarang berisi kolom points2 yang kita buat.

Contoh 2: Tetapkan beberapa variabel baru ke DataFrame

Kode berikut menunjukkan cara menggunakan metode penetapan() untuk menambahkan tiga variabel baru ke DataFrame:

 #add three new variables to DataFrame and store results in new DataFrame
df_new = df. assign (points2 = df. points * 2 ,
                   assists_rebs = df. assists + df. rebounds ,
                   conference = ' Western ')

#view new DataFrame
print (df_new)

   points assists rebounds points2 assists_rebs conference
0 25 5 11 50 16 Western
1 12 7 8 24 15 Western
2 15 7 10 30 17 Western
3 14 9 6 28 15 Western
4 19 12 6 38 18 Western
5 23 9 5 46 14 Western
6 25 9 9 50 18 Western
7 29 4 12 58 16 Western

Perhatikan bahwa tiga kolom baru telah ditambahkan ke DataFrame.

Catatan : Anda dapat menemukan dokumentasi lengkap metode pandas tugas() di sini .

Sumber daya tambahan

Tutorial berikut menjelaskan cara menggunakan fungsi umum lainnya di panda:

Cara menggunakan fungsi deskripsikan() di Pandas
Cara menggunakan fungsi idxmax() di Pandas
Cara menerapkan fungsi ke kolom yang dipilih di Pandas