Bagaimana untuk Mencipta Plot Taburan dengan Warna Berbeza untuk Tahap Kategori dalam Python Menggunakan Matplotlib?

Melukis Plot Taburan dengan Warna Berbeza untuk Tahap Kategori dalam Python dengan Matplotlib

Dalam Matplotlib, perpustakaan Python untuk mencipta statik, animasi dan visualisasi interaktif dalam Python, anda boleh memplot plot serakan berbeza dengan warna berbeza untuk setiap tahap pembolehubah kategori dengan memanfaatkan hujah c fungsi plt.scatter.

<code class="python">import matplotlib.pyplot as plt

df = pd.DataFrame({'x': [1, 2, 3], 'y': [4, 5, 6], 'color': ['red', 'blue', 'green']})

colors = {'red': 'tab:red', 'blue': 'tab:blue', 'green': 'tab:green'}

plt.scatter(df['x'], df['y'], c=df['color'].map(colors))
plt.show()</code>

Dengan menghantar hujah c, kamus pemetaan nama warna kepada nilai RGB boleh digunakan untuk menentukan warna setiap titik. Kaedah peta Pandas kemudian menggunakan pemetaan warna pada lajur df['color'], dengan berkesan memberikan setiap titik warna unik.

<code class="python">colors = {'D': 'tab:blue', 'E': 'tab:orange', 'F': 'tab:green', 'G': 'tab:red', 'H': 'tab:purple', 'I': 'tab:brown', 'J': 'tab:pink'}

ax.scatter(df['carat'], df['price'], c=df['color'].map(colors))</code>

Pendekatan ini membolehkan skema warna yang lebih disesuaikan dan kawalan yang lebih besar terhadap warna yang digunakan dalam plot. Dengan menggunakan kamus warna, pengguna boleh mengubah suai skema warna dengan mudah mengikut keperluan.

Atas ialah kandungan terperinci Bagaimana untuk Mencipta Plot Taburan dengan Warna Berbeza untuk Tahap Kategori dalam Python Menggunakan Matplotlib?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!