Menambahkan Legend dengan Kotak Warna pada Plot Matplotlib
Dalam artikel ini, kita akan belajar cara menambahkan legend (keterangan) dengan kotak warna pada plot menggunakan Matplotlib. Legend adalah area dalam plot yang menjelaskan elemen-elemen present di dalam grafik. Matplotlib menyediakan metode inbuilt bernama legend() untuk tujuan ini.
Contoh 1: Menambahkan Legend Sederhana
Berikut contoh sederhana cara menambahkan legend pada plot:
import matplotlib.pyplot as plt
# Membuat plot
plt.plot([1, 2, 3, 4])
# Menambahkan title
plt.title('Simple Legend')
# Menambahkan legend
plt.legend()
# Menampilkan plot
plt.show()
Dalam contoh di atas, kita menggunakan metode legend() untuk menambahkan legend pada plot. Legend ini akan muncul sebagai kotak warna yang menjelaskan elemen-elemen present di dalam grafik.
Contoh 2: Menambahkan Title ke Legend
Berikut contoh cara menambahkan title ke legend:
import matplotlib.pyplot as plt
# Membuat data
x = [3, 4, 4, 6, 8, 9]
y = [12, 14, 17, 16, 11, 13]
# Menambahkan legend dengan title
plt.scatter(x, y)
plt.title('Scatterplot Legend')
plt.legend(title='Class')
# Menampilkan plot
plt.show()
Dalam contoh di atas, kita menambahkan title ke legend menggunakan parameter title dalam metode legend().
Menambahkan Legend ke Scatterplot dengan PyQtGraph
Berikut contoh cara menambahkan legend ke scatterplot menggunakan PyQtGraph:
import sys
from PyQt5.QtWidgets import QApplication, QWidget
from PyQt5.QtGui import QPainter, QColor
from PyQt5.QtCore import Qt
class ScatterPlot(QWidget):
def __init__(self):
super().__init__()
# Membuat data
self.x = [3, 4, 4, 6, 8, 9]
self.y = [12, 14, 17, 16, 11, 13]
# Menambahkan scatterplot
self.scatter = pg.PlotDataItem(pos=self.x, neg=self.y)
def paintEvent(self, event):
qp = QPainter()
qp.begin(self)
self.drawScatter(qp)
qp.end()
def drawScatter(self, qp):
# Membuat brush untuk scatterplot
brush = QColor('red')
# Menambahkan scatterplot ke plot
self.scatter.setBrush(brush)
# Menampilkan legend
legend = pg.LegendItem()
self.scatter.addLegend()
Dalam contoh di atas, kita menggunakan PyQtGraph untuk menambahkan legend ke scatterplot. Legend ini akan muncul sebagai kotak warna yang menjelaskan elemen-elemen present di dalam grafik.
Menambahkan Legend ke Scatterplot dengan Matplotlib
Berikut contoh cara menambahkan legend ke scatterplot menggunakan Matplotlib:
import matplotlib.pyplot as plt
# Membuat data
x = [3, 4, 4, 6, 8, 9]
y = [12, 14, 17, 16, 11, 13]
# Menambahkan scatterplot
plt.scatter(x, y)
# Menambahkan legend
plt.legend(*scatter.legend_elements())
# Menampilkan plot
plt.show()
Dalam contoh di atas, kita menggunakan metode legend() untuk menambahkan legend ke scatterplot. Legend ini akan muncul sebagai kotak warna yang menjelaskan elemen-elemen present di dalam grafik.
Sumber Daya Tambahan
- Bagaimana Cara Menambahkan Ukuran Plot pada Matplotlib?
- Bagaimana Cara Mengatur Posisi Title pada Matplotlib?
- Bagaimana Cara Mengatur Rangkaian Axis pada Matplotlib?