Dalam analisis data, membuat grafik scatter yang memperlihatkan hubungan antara dua variabel kuantitatif adalah salah satu cara untuk melihat pola dan pattern dalam data. Namun, jika Anda memiliki lebih dari satu grup data yang ingin Anda visualisasikan, maka Anda perlu menambahkan legend pada grafik tersebut.

Pada artikel ini, kita akan membahas beberapa cara membuat grafik scatter dengan legend di Python menggunakan library matplotlib dan seaborn.

Membuat Grafik Scatter dengan Legend Menggunakan Matplotlib

Cara pertama untuk membuat grafik scatter dengan legend adalah menggunakan library matplotlib. Berikut adalah contoh kode:

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

df = pd.DataFrame(np.random.normal(10,1,30).reshape(10,3), 
 index = pd.date_range('2010-01-01', freq = 'M', periods = 10), 
 columns = ('one', 'two', 'three'))
df['key1'] = (4,4,4,6,6,6,8,8,8,8)

fig, ax = plt.subplots()
sc = ax.scatter(df['one'], df['two'], marker = 'o', c = df['key1'], alpha = 0.8)
ax.legend(*sc.legend_elements())
plt.show()

Pada kode di atas, kita menggunakan library numpy dan pandas untuk membuat data frame, kemudian menggunakan library matplotlib untuk membuat grafik scatter dengan legend.

Membuat Grafik Scatter dengan Legend Menggunakan Seaborn

Cara kedua untuk membuat grafik scatter dengan legend adalah menggunakan library seaborn. Berikut adalah contoh kode:

import seaborn as sns
sns.set()

df = pd.DataFrame(np.random.normal(10,1,30).reshape(10,3), 
 index = pd.date_range('2010-01-01', freq = 'M', periods = 10), 
 columns = ('one', 'two', 'three'))
df['key1'] = (4,4,4,6,6,6,8,8,8,8)

sns.scatterplot(data = df, x = 'one', y = 'two', hue = 'key1')
plt.show()

Pada kode di atas, kita menggunakan library seaborn untuk membuat grafik scatter dengan legend.

Membuat Grafik Scatter dengan Legend Menggunakan Altair

Cara ketiga untuk membuat grafik scatter dengan legend adalah menggunakan library Altair. Berikut adalah contoh kode:

import altair as alt

df = pd.DataFrame(np.random.normal(10,1,30).reshape(10,3), 
 index = pd.date_range('2010-01-01', freq = 'M', periods = 10), 
 columns = ('one', 'two', 'three'))
df['key1'] = (4,4,4,6,6,6,8,8,8,8)

c = alt.Chart(df).mark_circle().encode(x='one', y='two', color='key1')
c

Pada kode di atas, kita menggunakan library Altair untuk membuat grafik scatter dengan legend.

Dalam kesimpulan, kita telah membahas beberapa cara membuat grafik scatter dengan legend di Python menggunakan library matplotlib, seaborn, dan Altair. Kita dapat memilih cara yang sesuai dengan kebutuhan kami.