要自定義顏色映射(cmap)可以使用matplotlib庫中的colors模塊。下面是一個示例代碼,展示如何自定義顏色映射:
import matplotlib.pyplot as plt
import numpy as np
from matplotlib.colors import LinearSegmentedColormap
# 自定義顏色映射
colors = [(0, 0, 0), (1, 1, 1), (0.5, 0, 0.5)] # RGB顏色值
cmap_name = 'custom_cmap'
cm = LinearSegmentedColormap.from_list(cmap_name, colors, N=100)
# 生成數據
data = np.random.rand(10, 10)
# 繪制圖像
plt.imshow(data, cmap=cm)
plt.colorbar()
plt.show()
在這個示例中,我們首先定義了一個顏色列表colors
,其中包含了3種顏色的RGB值。然后使用LinearSegmentedColormap.from_list
方法創建了一個自定義的顏色映射cm
。接著我們生成了一組隨機數據并使用這個自定義的顏色映射繪制了圖像。
你可以根據需要調整顏色列表colors
來自定義顏色映射。