python cmap如何自定義顏色映射

小樊
163
2024-06-15 12:35:28

要自定義顏色映射（cmap）可以使用matplotlib庫(kù)中的colors模塊。下面是一個(gè)示例代碼，展示如何自定義顏色映射：

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)

# 生成數(shù)據(jù)
data = np.random.rand(10, 10)

# 繪制圖像
plt.imshow(data, cmap=cm)
plt.colorbar()
plt.show()

在這個(gè)示例中，我們首先定義了一個(gè)顏色列表colors，其中包含了3種顏色的RGB值。然后使用LinearSegmentedColormap.from_list方法創(chuàng)建了一個(gè)自定義的顏色映射cm。接著我們生成了一組隨機(jī)數(shù)據(jù)并使用這個(gè)自定義的顏色映射繪制了圖像。

你可以根據(jù)需要調(diào)整顏色列表colors來(lái)自定義顏色映射。

0 踩