Menggambar Grafik Scatter dengan Menggunakan Data Frame
Pandas menyediakan berbagai cara untuk menggambar grafik, termasuk scatter plot. Dalam artikel ini, kita akan belajar bagaimana menggunakan DataFrame.plot.scatter untuk menggambar grafik scatter dengan ukuran dan warna yang dapat diubah.
Parameter
Fungsi DataFrame.plot.scatter memiliki beberapa parameter yang perlu diatur sebelum menggambar grafik:
x: Nama kolom atau posisi kolumnya yang akan digunakan sebagai koordinat horizontal untuk masing-masing titik.y: Nama kolom atau posisi kolumnya yang akan digunakan sebagai koordinat vertikal untuk masing-masing titik.s: Ukuran masing-masing titik. Bisa berupa:
- String nama kolom yang akan digunakan untuk ukuran marker.
- Skalar tunggal, sehingga semua titik memiliki ukuran sama.
- Sebuah sequence skalar, yang akan digunakan secara rekursif untuk setiap titik's size.
c: Warna masing-masing titik. Bisa berupa:
- String warna nama, kode RGB atau RGBA, contohnya 'red' atau '#a98d19'.
- Sebuah sequence string warna nama, kode RGB atau RGBA, yang akan digunakan secara rekursif untuk setiap titik's color.
- Nama kolom atau posisi kolumnya yang akan digunakan untuk menentukan warna marker points berdasarkan colormap.
Contoh
Mari lihat contoh bagaimana kita dapat menggambar grafik scatter menggunakan koordinat dari nilai-nilai dalam kolom-kolom DataFrame:
import pandas as pd
df = pd.DataFrame([[5.1, 3.5, 0], [4.9, 3.0, 0], [7.0, 3.2, 1],
[6.4, 3.2, 1], [5.9, 3.0, 2]],
columns=['length', 'width', 'species'])
ax1 = df.plot.scatter(x='length',
y='width',
c='DarkBlue')
ax2 = df.plot.scatter(x='length',
y='width',
c='species',
colormap='viridis')
Dalam contoh di atas, kita menggunakan koordinat dari kolom 'length' dan 'width' sebagai koordinat horizontal dan vertikal, serta warna 'DarkBlue' untuk semua titik. Dalam contoh kedua, kita menggunakan koordinat yang sama, tetapi warna tiap-titik ditentukan oleh nilai-nilai dalam kolom 'species', dengan menggunakan colormap 'viridis'.