Menggambar Scatter Plot dengan Matplotlib
Matplotlib adalah salah satu library Python yang paling populer digunakan untuk visualisasi data. Dalam artikel ini, kita akan membahas cara menggambar scatter plot menggunakan Matplotlib.
1. Instalasi Matplotlib
Sebelum kita dapat menggunakan Matplotlib, kita harus terlebih dahulu menginstalnya. Anda dapat mengunduh dan menginstal Matplotlib melalui pip dengan perintah berikut:
pip install matplotlib
2. Menggambar Scatter Plot dengan Density
Kita dapat menggunakan fungsi hist2d
dari Matplotlib untuk menggambar scatter plot dengan density. Contoh kode sebagai berikut:
import numpy as np
import matplotlib.pyplot as plt
a = np.random.normal(size=1000)
b = a*3 + np.random.normal(size=1000)
plt.hist2d(a, b, (50, 50), cmap=plt.cm.jet)
plt.colorbar()
Kode di atas akan menggambar scatter plot dengan density yang ditampilkan dalam bentuk warna.
3. Menggambar Scatter Plot dengan Trend Line
Kita juga dapat menggunakan fungsi polyfit
dan plot
dari Matplotlib untuk menggambar trend line pada scatter plot. Contoh kode sebagai berikut:
import numpy as np
import matplotlib.pyplot as plt
x = np.array([1, 1, 2, 3, 4, 4, 5, 6, 7, 7, 8, 9])
y = np.array([13, 14, 17, 12, 23, 24, 25, 25, 24, 28, 32, 33])
plt.plot(x, y, 'o')
m, b = np.polyfit(x, y, 1)
plt.plot(x, m*x+b)
Kode di atas akan menggambar scatter plot dengan trend line yang ditampilkan dalam bentuk garis.
Menggambar Scatter Plot pada 2 Y-Axis dengan Matplotlib
Jika Anda ingin menggambar scatter plot pada 2 y-axis dengan Matplotlib, maka Anda dapat menggunakan fungsi twinx
dan plot
dari Matplotlib. Contoh kode sebagai berikut:
import matplotlib.pyplot as plt
fig, ax1 = plt.subplots()
ax2 = ax1.twinx()
ax1.plot(epoch, list_of_gas, 'b')
ax2.plot(temp_times, temperatures, 'r')
ax1.set_ylabel('Pressure (torr)', color='b')
ax2.set_ylabel('Temperature (Celcius)', color='r')
ax1.set_title(folder + ' - ' + gas)
ax1.set_xlabel('time (seconds)')
ax1.set_xlim([0, 1000000])
ax2.set_ylim([0,425])
ax1.set_yscale('log')
ax1.set_ylim([ymin,ymax])
plt.show()
Kode di atas akan menggambar scatter plot pada 2 y-axis dengan menggunakan fungsi twinx
dan plot
. Namun, jika Anda ingin menghapus garis yang terhubung antara titik-titik, maka Anda dapat menggunakan fungsi scatter
dari Matplotlib.
Kesimpulan
Dalam artikel ini, kita telah membahas cara menggambar scatter plot dengan Matplotlib. Kita juga telah membahas cara menggambar trend line dan menghapus garis yang terhubung antara titik-titik pada 2 y-axis. Dengan menggunakan Matplotlib, Anda dapat dengan mudah menciptakan visualisasi data yang indah dan informatif.