在Python中如何自定義colormap

小樊
161
2024-06-28 23:11:28

在Python中，可以使用matplotlib庫(kù)中的colors模塊來(lái)自定義colormap。以下是一個(gè)簡(jiǎn)單的示例：

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

# 定義顏色映射
colors = [(0, 0, 0), (0.5, 0.5, 0.5), (1, 1, 1)]  # 定義三種顏色，分別為黑色、灰色和白色
cmap_name = 'custom_colormap'
cm = LinearSegmentedColormap.from_list(cmap_name, colors, N=100)

# 創(chuàng)建一張圖片
data = np.random.rand(10, 10)
plt.imshow(data, cmap=cm)
plt.colorbar()
plt.show()

在這個(gè)例子中，我們首先定義了一個(gè)包含三種顏色的顏色列表colors，然后使用LinearSegmentedColormap.from_list()方法創(chuàng)建了一個(gè)自定義的colormap，并命名為’custom_colormap’。最后，我們將這個(gè)自定義的colormap應(yīng)用到了一張隨機(jī)生成的圖片上，并顯示出來(lái)。

0 踩