python画图指定colormap中间隔取样，并转为列表形式画图

原创于 2023-05-15 10:56:40 发布 · 497 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #matplotlib #开发语言

Python从入门到入土专栏收录该内容

23 篇文章

订阅专栏

文章展示了如何使用numpy和matplotlib库来创建自定义的颜色映射。首先从plasmacolormap中获取256个颜色，然后选取20个等间隔的整数生成新的颜色列表。通过ListedColormap创建新colormap，并用这些颜色绘制一系列矩形以展示新颜色集。

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

cmap_256 = plt.get_cmap('plasma', 256)  # 获取长度为256的colormap
colors = np.linspace(0, 255, 20)  # 生成20个在0-255之间等间隔的整数
new_cmap = cmap_256(colors.astype(int))  # 将这20个整数对应的颜色作为新的colormap的颜色值
new_cmap = ListedColormap(new_cmap)  # 创建新的colormap

colors = [new_cmap(i) for i in range(new_cmap.N)]
print(colors)

# 绘制一个包含此颜色的矩形
for i in range(len(colors)):
    fig, ax = plt.subplots()
    ax.add_patch(plt.Rectangle((0, 0), 1, 1, color=colors[i]))
    plt.show()