Konversi Pandas DataFrame menjadi Numpy Array

Saat bekerja dengan pandas DataFrames, terkadang bermanfaat jika kita mengubahnya menjadi array NumPy. Yang terakhir tampaknya lebih hemat memori, terutama ketika harus melakukan beberapa operasi matematika yang kompleks terhadap data.

Ini terutama benar ketika kita bekerja dengan jumlah data yang relatif kecil (katakanlah 50 ribu row atau kurang). Pandas biasanya akan mengungguli arrays dalam kasus yang melibatkan volume data yang jauh lebih besar (misalnya >500 ribu baris). Ini hanya aturan praktis — dalam sebagian besar kasus, akan lebih baik untuk menguji kedua opsi dan melihat mana yang lebih baik dalam hal kinerja dan penggunaan memori, berdasarkan kebutuhan spesifik dan kasus penggunaan Anda.

Dalam tutorial singkat hari ini kita akan melakukan cara mengonversi pandas DataFrame menjadi array NumPy secara efisien.

Pertama, mari buat contoh pandas DataFrame yang akan kita gunakan untuk mendemonstrasikan beberapa cara berbeda yang berpotensi digunakan untuk mengubahnya menjadi array numpy.

Code diatas digunakan untuk membuat dataframe yang mempunya 4 kolom. Kemudian di bawah ini adalah dataframe yang berhasil kita buat

Menggunakan pandas.DataFrame.to_numpy()

Opsi pertama yang kita miliki saat mengubah pandas DataFrame menjadi array NumPy adalah fungsi .to_numpy() . Dibawah ini ditunjukkan penggunakan fungsi .to_numpy().

Dan kita bisa cek jenis objek yaitu adalah numpy.ndarray: 

Menggunakan pandas.DatFrame.to_records()

Opsi lain di sini adalah fungsi .to_records() yang akan mengonversi pandas DataFrame menjadi array record NumPy:

Seperti disebutkan, berbeda dengan to_numpy(), fungsi to_records() akan mengembalikan objek bertipe nympy.recarray:

Menggunakan numpy.asarray()

Opsi ketiga yang kita miliki di sini adalah Fungsi numpy.asarray() yang akan mengubah input pandas DataFrame menjadi array NumPy:

Objek yang kembali akan sekali lagi menjadi turunan dari numpy.ndarray:

Dalam postingan kali ini kita membahas tentang mengonversi panda DataFrames menjadi array NumPy dan dalam situasi apa hal itu dapat bermanfaat untuk dilakukan.

Selain itu, kami menunjukkan cara mengonversi DataFrame menjadi ndarray menggunakan fungsi to_numpy() dan to_records() dari objek pandas.DataFrame serta fungsi numpy.asarray().