Matplotlib adalah salah satu library Python yang paling populer digunakan untuk menganalisis dan menggambar data. Salah satu fitur yang sangat berguna dalam matplotlib adalah kemampuan untuk mengwarnai plot scatter berdasarkan nilai atau kategori. Dalam tutorial ini, kita akan belajar cara membuat plot scatter dengan warna yang berbeda untuk setiap titik berdasarkan nilai atau kategori.

Contoh 1: Mengwarnai Plot Scatter dengan Nilai

Supose kita memiliki DataFrame pandas sebagai berikut:

import pandas as pd

df = pd.DataFrame({'x': [25, 12, 15, 14, 19, 23, 25, 29],
 'y': [5, 7, 7, 9, 12, 9, 9, 4],
 'z': [3, 4, 4, 5, 7, 8, 8, 9]})

print(df)

Kita dapat membuat plot scatter dengan menggunakan sintaks berikut:

import matplotlib.pyplot as plt

plt.scatter(df.x, df.y, s=200, c=df.z, cmap='gray')

Dalam contoh ini, kita menggunakan nilai untuk variable z sebagai warna untuk plot scatter. Kita juga dapat mengganti warna dengan menggunakan colormap yang berbeda, seperti 'Greens' atau 'Blues'. Contoh berikut menunjukkan cara membuat plot scatter dengan warna hijau:

plt.scatter(df.x, df.y, s=200, c=df.z, cmap='Greens')

Contoh 2: Mengwarnai Plot Scatter dengan Kategori

Supose kita memiliki DataFrame pandas sebagai berikut:

import pandas as pd

df = pd.DataFrame({'x': [25, 12, 15, 14, 19, 23, 25, 29],
 'y': [5, 7, 7, 9, 12, 9, 9, 4],
 'z': ['A', 'A', 'B', 'B', 'B', 'C', 'C', 'C']})

print(df)

Kita dapat membuat plot scatter dengan menggunakan sintaks berikut:

import matplotlib.pyplot as plt

groups = df.groupby('z')
for name, group in groups:
 plt.plot(group.x, group.y, marker='o', linestyle='', markersize=12, label=name)

plt.legend()

Dalam contoh ini, kita menggunakan variable z untuk membagi data menjadi kategori dan kemudian membuat plot scatter yang berbeda untuk setiap kategori.

Dengan demikian, kita telah belajar cara membuat plot scatter dengan warna yang berbeda untuk setiap titik berdasarkan nilai atau kategori. Kita dapat menggunakan library matplotlib untuk membuat plot scatter yang lebih menarik dan interaktif.