Cara menggunakan DATAFRAME.COLUMNS pada Python

Mengubah nama column. Parameter yang digunakan adalah columns dengan argument Python Dictionary yang berisi nama kolom lama dan nama kolom baru.

Contoh mengganti nama kolom Year menjadi Tahun dan Actor menjadi Aktor.

df.rename(columns = {"Year": "Tahun", "Actor": "Aktor"})

# df.rename(columns = {"Year": "Tahun", "Actor": "Aktor"}, inplace=True)

Gunakan parameter inplace=True untuk mengaplikasikan permanent pada dataframe.

Cara kedua menggunakan perintah columns. Kita harus memasukan semua kolom termasuk yang tidak akan diubah.

#perintah ini akan error
df.columns = ["Tahun", "Aktor"]

#perintah ini akan mengubah kolom Year dan Actor, dan yang lainnya tetap.
df.columns = ["Tahun", "Aktor", "Director",	"Box Office",	"Budget", "Bond Actor Salary"]

Kekurangan dari cara diatas, kita kehilangan informasi nama kolom originalnya. Sementara dengan method rename, nama kolom sebelumnya masih bisa diketahui.

Dalam analisis data terkadang kita perlu membuat kolom baru untuk menampung informasi yang lebih detail yang dapat membantu kita melihat lebih jauh tentang data yang sedang kita observasi. Untuk itu, pada tulisan ini, saya akan membahas cara menambahkan kolom baru pada dataframe pandas.

Setidaknya ada 3 cara yang dapat kita lakukan untuk menambahkan kolom baru, baik itu menambahkan kolom baru di akhir dataframe ataupun menyisipkannya di antara kolom-kolom yang sudah ada.

Disini, contoh yang akan digunakan adalah bagaimana membuat kolom baru dari data pada kolom yang sudah ada. Misalnya dalam contoh kali ini, kita akan membuat dataframe df yang berisi data nilai mahasiswa seperti di bawah ini.

import pandas as pd

data = {'Nama': ['Amanda', 'Andi', 'Anton', 'Arini', 'Aziz'],
        'Tugas': [77, 80, 91, 83, 78],
        'UTS': [90, 76, 88, 69, 75],
        'UAS': [80, 87, 77, 90, 88]}

df = pd.DataFrame(data)
df

Baca Juga: Manipulasi Series Pandas dengan map() Python

Output:

Seperti yang dapat kita lihat, ada 4 kolom yaitu Nama, Tugas, UTS, dan UAS. Misalnya kita ingin menambahkan kolom baru yaitu Nilai Akhir yang berisi perhitungan akhir dari nilai Tugas, UTS, dan UAS.

Sekarang kita coba aplikasikan menggunakan cara-cara di bawah ini.

Cara sederhana, tanpa menggunakan method

Cara pertama yang akan kita bahas untuk menambahkan kolom baru pada dataframe Pandas adalah dengan langsung mendefiniskan kolom baru tanpa menggunakan method apapun. Perhatikan kode berikut.

import pandas as pd

data = {'Nama': ['Amanda', 'Andi', 'Anton', 'Arini', 'Aziz'],
        'Tugas': [77, 80, 91, 83, 78],
        'UTS': [90, 76, 88, 69, 75],
        'UAS': [80, 87, 77, 90, 88]}

df = pd.DataFrame(data)

# Menambahkan kolom baru
df['Nilai Akhir'] = df['Tugas']*0.3 + df['UTS']*0.3 + df['UAS']*0.4
df

Baca Juga: Cara Membuat Bar Plot dengan Matplotlib Python

Output:

Dari kode di atas, kita hanya perlu mendefinisikan

import pandas as pd

data = {'Nama': ['Amanda', 'Andi', 'Anton', 'Arini', 'Aziz'],
        'Tugas': [77, 80, 91, 83, 78],
        'UTS': [90, 76, 88, 69, 75],
        'UAS': [80, 87, 77, 90, 88]}

df = pd.DataFrame(data)

# Menambahkan kolom baru
df['Nilai Akhir'] = df['Tugas']*0.3 + df['UTS']*0.3 + df['UAS']*0.4
df

Menggunakan assign()

Cara kedua untuk menambahkan kolom baru pada dataframe adalah dengan menggunakan method

import pandas as pd

data = {'Nama': ['Amanda', 'Andi', 'Anton', 'Arini', 'Aziz'],
        'Tugas': [77, 80, 91, 83, 78],
        'UTS': [90, 76, 88, 69, 75],
        'UAS': [80, 87, 77, 90, 88]}

df = pd.DataFrame(data)

# Menambahkan kolom baru
df['Nilai Akhir'] = df['Tugas']*0.3 + df['UTS']*0.3 + df['UAS']*0.4
df

import pandas as pd

data = {'Nama': ['Amanda', 'Andi', 'Anton', 'Arini', 'Aziz'],
        'Tugas': [77, 80, 91, 83, 78],
        'UTS': [90, 76, 88, 69, 75],
        'UAS': [80, 87, 77, 90, 88]}

df = pd.DataFrame(data)

# Menambahkan kolom baru dengan assign
df_akhir = df.assign(Nilai_Akhir = df['Tugas']*0.3 + df['UTS']*0.3 + df['UAS']*0.4)
df_akhir

Baca Juga: 2 Teknik Reduksi Dimensi Populer dengan Python

Output:

Seperti yang dapat kita lihat, outputnya tidak jauh berbeda dengan menggunakan cara sederhana. Hanya saja penamaan kolomnya tidak boleh mengandung spasi, jadi disini saya beri nama kolomnya

import pandas as pd

data = {'Nama': ['Amanda', 'Andi', 'Anton', 'Arini', 'Aziz'],
        'Tugas': [77, 80, 91, 83, 78],
        'UTS': [90, 76, 88, 69, 75],
        'UAS': [80, 87, 77, 90, 88]}

df = pd.DataFrame(data)

# Menambahkan kolom baru
df['Nilai Akhir'] = df['Tugas']*0.3 + df['UTS']*0.3 + df['UAS']*0.4
df

Menggunakan insert()

Nah, jika kita ingin menyisipkan kolom baru di antara kolom-kolom yang sudah ada, maka kita dapat menggunakan method

import pandas as pd

data = {'Nama': ['Amanda', 'Andi', 'Anton', 'Arini', 'Aziz'],
        'Tugas': [77, 80, 91, 83, 78],
        'UTS': [90, 76, 88, 69, 75],
        'UAS': [80, 87, 77, 90, 88]}

df = pd.DataFrame(data)

# Menambahkan kolom baru
df['Nilai Akhir'] = df['Tugas']*0.3 + df['UTS']*0.3 + df['UAS']*0.4
df

import pandas as pd

data = {'Nama': ['Amanda', 'Andi', 'Anton', 'Arini', 'Aziz'],
        'Tugas': [77, 80, 91, 83, 78],
        'UTS': [90, 76, 88, 69, 75],
        'UAS': [80, 87, 77, 90, 88]}

df = pd.DataFrame(data)

# Menambahkan kolom baru
df['Nilai Akhir'] = df['Tugas']*0.3 + df['UTS']*0.3 + df['UAS']*0.4
df

Langsung saja kita lihat contoh kodenya.

import pandas as pd

data = {'Nama': ['Amanda', 'Andi', 'Anton', 'Arini', 'Aziz'],
        'Tugas': [77, 80, 91, 83, 78],
        'UTS': [90, 76, 88, 69, 75],
        'UAS': [80, 87, 77, 90, 88]}

df = pd.DataFrame(data)

# Menambahkan kolom baru dengan insert
df.insert(1, 'Nilai Akhir', df['Tugas']*0.3 + df['UTS']*0.3 + df['UAS']*0.4)
df

Output:

Dapat dilihat dari output di atas, kolom Nilai Akhir yang kita tambahkan pada posisi indeks 1 di dataframe berada setelah kolom Nama karena penghitungan indeks kolom di Pandas dimulai dari 0.

Itulah 3 cara yang dapat dilakukan untuk menambahkan kolom baru pada dataframe Pandas yang dihasilkan dari data pada kolom-kolom yang sudah ada.

Apa itu DataFrame pada Python?

Apa fungsi yang digunakan untuk menampilkan jumlah baris dan kolom pada sebuah data frame?

Apa yang dimaksud dengan Pandas?

Apa bentuk data yang ada dalam Pandas?