Cara memasukkan kolom ke dalam pandas dataframe

Seringkali Anda mungkin ingin memasukkan kolom baru ke dalam pandas DataFrame. Untungnya, hal ini mudah dilakukan dengan menggunakan fungsi pandas insert() , yang menggunakan sintaks berikut:

masukkan(lokasi, kolom, nilai, izinkan_duplikat=False)

Emas:

• loc: indeks untuk menyisipkan kolom. Kolom pertama adalah 0.
• kolom: Nama untuk diberikan ke kolom baru.
• value: Array nilai untuk kolom baru.
• izinkan_duplikat: Mengizinkan atau tidak nama kolom baru cocok dengan nama kolom yang ada. Nilai defaultnya salah.

Tutorial ini menunjukkan beberapa contoh penggunaan praktis fungsi ini.

Contoh 1: Sisipkan kolom baru sebagai kolom pertama

Kode berikut menunjukkan cara menyisipkan kolom baru sebagai kolom pertama dari DataFrame yang sudah ada:

 import pandas as pd

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

#view DataFrame
df
        points assists rebounds
0 25 5 11
1 12 7 8
2 15 7 10
3 14 9 6
4 19 12 6

#insert new column 'player' as first column
player_vals = ['A', 'B', 'C', 'D', 'E']
df. insert (loc= 0 , column='player', value=player_vals)
df

        player points assists rebounds
0 to 25 5 11
1 B 12 7 8
2 C 15 7 10
3 D 14 9 6
4 E 19 12 6

Contoh 2: Sisipkan kolom baru sebagai kolom tengah

Kode berikut menunjukkan cara menyisipkan kolom baru sebagai kolom ketiga dari DataFrame yang ada:

 import pandas as pd

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

#insert new column 'player' as third column
player_vals = ['A', 'B', 'C', 'D', 'E']
df. insert (loc= 2 , column='player', value=player_vals)
df

        points assists player rebounds
0 25 5 A 11
1 12 7 B 8
2 15 7 C 10
3 14 9 D 6
4 19 12 E 6

Contoh 3: Sisipkan kolom baru sebagai kolom terakhir

Kode berikut menunjukkan cara menyisipkan kolom baru sebagai kolom terakhir dari DataFrame yang ada:

 import pandas as pd

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

#insert new column 'player' as last column
player_vals = ['A', 'B', 'C', 'D', 'E']
df. insert (loc= len (df.columns) , column='player', value=player_vals)
df

        points assists player rebounds
0 25 5 A 11
1 12 7 B 8
2 15 7 C 10
3 14 9 D 6
4 19 12 E 6

Perhatikan bahwa menggunakan len(df.columns) memungkinkan Anda menyisipkan kolom baru sebagai kolom terakhir dari dataFrame apa pun, terlepas dari berapa banyak kolom yang dikandungnya.

Anda dapat menemukan dokumentasi lengkap fungsi insert() di sini .