python画图指定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()