Rumah  >  Artikel  >  pembangunan bahagian belakang  >  Bagaimanakah anda boleh mencipta bar warna diskret tersuai untuk scatterplot dalam Matplotlib?

Bagaimanakah anda boleh mencipta bar warna diskret tersuai untuk scatterplot dalam Matplotlib?

Patricia Arquette
Patricia Arquetteasal
2024-10-25 06:37:02561semak imbas

How can you create a custom discrete colorbar for a scatterplot in Matplotlib?

Mendiskritkan Bar Warna Matplotlib untuk Scatterplot

Dalam plot serakan, bar warna berterusan mungkin tidak sentiasa mencukupi untuk mewakili data diskret. Mencipta bar warna diskret adalah penting untuk menggambarkan corak asas secara berkesan.

Untuk mencapai matlamat ini, kelas BoundaryNorm dalam Matplotlib boleh dimanfaatkan sebagai penormal untuk plot serakan. Ini memastikan bahawa nilai yang berbeza diwakili oleh warna yang unik.

Satu cabaran dalam menyesuaikan bar warna timbul apabila cuba menetapkan nilai tertentu untuk dipaparkan sebagai kelabu. Untuk menangani isu ini, palet warna sedia ada boleh diubah suai dengan mengekstrak dan mengatasi entri warna.

<code class="python">cmaplist = [cmap(i) for i in range(cmap.N)]
# force the first color entry to be grey
cmaplist[0] = (.5, .5, .5, 1.0)</code>

Selepas mengubah suai palet warna, peta warna tersuai boleh dibuat. BoundaryNorm kemudiannya digunakan untuk mentakrifkan binning dan menormalkan data dengan sewajarnya.

<code class="python">cmap = mpl.colors.LinearSegmentedColormap.from_list(
    'Custom cmap', cmaplist, cmap.N)
bounds = np.linspace(0, 20, 21)
norm = mpl.colors.BoundaryNorm(bounds, cmap.N)</code>

Dengan palet warna yang diubah suai dan penormalan di tempatnya, plot serakan boleh dipaparkan. Paksi berasingan ditambahkan untuk menampung bar warna, yang disesuaikan untuk memaparkan sempadan diskret dan warna sepadannya.

<code class="python">scat = ax.scatter(x, y, c=tag, cmap=cmap, norm=norm)
ax2 = fig.add_axes([0.95, 0.1, 0.03, 0.8])
cb = plt.colorbar.ColorbarBase(ax2, cmap=cmap, norm=norm,
    spacing='proportional', ticks=bounds, boundaries=bounds, format='%1i')</code>

Pendekatan ini membolehkan penciptaan bar warna diskret tersuai yang menyampaikan data diskret secara berkesan dalam plot serakan, memberikan gambaran visual yang lebih jelas bagi corak asas.

Atas ialah kandungan terperinci Bagaimanakah anda boleh mencipta bar warna diskret tersuai untuk scatterplot dalam Matplotlib?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!

Kenyataan:
Kandungan artikel ini disumbangkan secara sukarela oleh netizen, dan hak cipta adalah milik pengarang asal. Laman web ini tidak memikul tanggungjawab undang-undang yang sepadan. Jika anda menemui sebarang kandungan yang disyaki plagiarisme atau pelanggaran, sila hubungi admin@php.cn