Dalam artikel ini, kita akan membahas cara menambahkan garis ke grafik scatter menggunakan Python dan bibliothek Matplotlib dan Seaborn. Kita juga akan melihat contoh-contoh penggunaan yang lebih lanjut.

Prasyarat

Untuk mengikuti panduan ini, Anda harus memiliki pemahaman dasar tentang programing Python dan memiliki Python terinstall di sistem Anda. Selain itu, Anda perlu menginstal bibliothek-bibliothek berikut jika belum terinstall:

  • Matplotlib
  • Seaborn
  • NumPy (opsional untuk generasi data)

Anda dapat menginstal bibliothek-bibliothek tersebut menggunakan pip:

pip install matplotlib seaborn numpy

3. Membuat Grafik Scatter

Mari kita mulai dengan membuat grafik scatter sederhana menggunakan Matplotlib dan Seaborn.

3.1. Menghasilkan Data Sampel

Kita akan menghasilkan data sampel menggunakan NumPy:

import numpy as np

# Generate random data
np.random.seed(0)
x = np.random.rand(50)
y = 2 * x + np.random.normal(0, 0.1, 50)

3.2. Plotting dengan Matplotlib

Sekarang, mari kita membuat grafik scatter dengan Matplotlib:

import matplotlib.pyplot as plt

# Create a scatter plot
plt.scatter(x, y, color='blue')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.title('Scatter Plot with Matplotlib')
plt.show()

3.3. Plotting dengan Seaborn

Sama-sama, kita dapat membuat grafik scatter dengan Seaborn:

import seaborn as sns

# Create a scatter plot
sns.scatterplot(x=x, y=y)
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.title('Scatter Plot with Seaborn')
plt.show()

4. Menambahkan Garis ke Grafik Scatter

Menambahkan garis ke grafik scatter dapat dilakukan untuk berbagai tujuan, seperti menghitung garis terbaik, menambahkan garis referensi, atau menunjukkan batas-batas.

4.1. Menambahkan Garis Terbaik

Contoh penggunaan yang paling umum adalah menambahkan garis terbaik, yang membantu memvisualisasikan hubungan antara variabel-variabel.

Dengan Matplotlib

Untuk menambahkan garis terbaik dengan Matplotlib, kita dapat menggunakan NumPy untuk menghitung garis:

# Calculate the line of best fit
slope, intercept = np.polyfit(x, y, 1)
line = slope * x + intercept

# Plot the scatter plot and line of best fit
plt.scatter(x, y, color='blue')
plt.plot(x, line, color='red', label='Line of Best Fit')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.title('Scatter Plot with Line of Best Fit')
plt.legend()
plt.show()

Dengan Seaborn

Seaborn memudahkan proses ini dengan fungsi regplot:

# Plot the scatter plot and line of best fit
sns.regplot(x=x, y=y, ci=None, line_kws={"color": "red"})
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.title('Scatter Plot with Line of Best Fit')
plt.show()

4.2. Menambahkan Garis Kustom

Kita mungkin juga ingin menambahkan garis kustom, seperti garis horizontal atau vertical, atau garis dengan kemiringan dan intercept tertentu.

4.2.1. Garis Horizontal dan Vertical

Menambahkan garis horizontal dan vertical dengan Matplotlib:

# Plot the scatter plot
plt.scatter(x, y, color='blue')

# Add a horizontal line at y = 1
plt.axhline(y=1, color='green', linestyle='--', label='y=1')

# Add a vertical line at x = 0.5
plt.axvline(x=0.5, color='purple', linestyle='--', label='x=0.5')

plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.title('Scatter Plot with Custom Lines')
plt.legend()
plt.show()

4.2.2. Garis dengan Kemiringan dan Intercept Kustom

Menambahkan garis dengan kemiringan dan intercept kustom:

# Define the slope and intercept
slope = 1.5
intercept = -2

# Plot the scatter plot
plt.scatter(x, y, color='blue')

# Add a line with custom slope and intercept
plt.plot(x, slope * x + intercept, color='red', label='Custom Line')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.title('Scatter Plot with Custom Lines')
plt.legend()
plt.show()

Dalam artikel ini, kita telah membahas cara menambahkan garis ke grafik scatter menggunakan Python dan bibliothek Matplotlib dan Seaborn. Kita juga telah melihat contoh-contoh penggunaan yang lebih lanjut.