首頁  >  文章  >  後端開發  >  如何在 Matplotlib 中建立自訂色彩圖並新增色標?

如何在 Matplotlib 中建立自訂色彩圖並新增色標?

Susan Sarandon
Susan Sarandon原創
2024-11-13 05:24:02405瀏覽

How to Create a Custom Colormap and Add a Color Scale in Matplotlib?

建立自訂顏色圖並合併色標

要建立自己的顏色圖,一種方法是利用 matplotlib.colors 模組中的 LinearSegmentedColormap 函數。這種方法更簡單,並產生連續的色標。

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()

此程式碼片段成功地建立了一個從紅色到紫色到藍色平滑過渡的顏色圖,範圍從 -2 到 2。色標也是合併到圖的右側,可以輕鬆解釋顏色。

以上是如何在 Matplotlib 中建立自訂色彩圖並新增色標?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn