Bagaimana untuk Mencipta Peta Warna Tersuai dan Menambah Skala Warna dalam Matplotlib?

Mencipta Peta Warna Tersuai dan Menggabungkan Skala Warna

Untuk mencipta peta warna anda sendiri, satu pendekatan ialah menggunakan fungsi LinearSegmentedColormap daripada modul matplotlib.colors. Pendekatan ini lebih mudah dan menghasilkan skala warna yang berterusan.

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.colors

# Generate random data points
x, y, c = zip(*np.random.rand(30, 3) * 4 - 2)

# Define lower and upper bounds for normalization
norm = plt.Normalize(-2, 2)

# Create a list of tuples representing the values and corresponding colors
tuples = [(norm(-2.), 'red'), (norm(-1.), 'violet'), (norm(2.), 'blue')]

# Generate the colormap from the list of tuples
cmap = matplotlib.colors.LinearSegmentedColormap.from_list('', tuples)

# Plot the data points using the custom colormap
plt.scatter(x, y, c=c, cmap=cmap, norm=norm)

# Add a color scale to the plot
plt.colorbar()
plt.show()

Coretan kod ini berjaya mencipta peta warna dengan peralihan lancar daripada merah kepada ungu kepada biru, antara -2 hingga 2. Skala warna juga digabungkan di sebelah kanan plot, membolehkan tafsiran warna yang mudah.

Atas ialah kandungan terperinci Bagaimana untuk Mencipta Peta Warna Tersuai dan Menambah Skala Warna dalam Matplotlib?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!