如何从色阶截图生成Matplotlib Colormap对象？

嘿，你的思路其实已经找对方向了！从色阶截图提取颜色来生成自定义Colormap完全可行，我来帮你把代码补全，再给几个更顺手的优化方案：

1. 补全你的初始实现（最直接的方式） #

你已经用PIL打开了图片，接下来只需要提取色阶的颜色序列，再用Matplotlib的工具生成Colormap就行。这里我把代码补全并优化了一下：

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

# 加载色阶截图
img = Image.open('pet_colourbar.png')
# 转为numpy数组并将像素值缩放到0-1区间（Matplotlib要求的颜色范围）
img_array = np.array(img) / 255.0

# 提取色阶颜色：假设你的色阶是垂直方向（从上到下渐变），取第一列所有像素
# 如果是水平色阶（从左到右渐变），就改成取第一行：img_array[0, :, :]
rgb_colours = img_array[:, 0, :]

# 生成自定义Colormap
custom_cmap = LinearSegmentedColormap.from_list(
    'pet_custom_cmap',  # Colormap的名字，随便取
    rgb_colours,
    N=len(rgb_colours)  # 颜色的数量，默认是256，这里用截图的像素数更精准
)

# 测试一下效果
plt.imshow(np.random.rand(10, 10), cmap=custom_cmap)
plt.colorbar()
plt.show()

2. 自动适配色阶方向（不用手动判断） #

如果不确定截图的色阶是水平还是垂直，可以加个简单的判断，让代码自动识别：

# 判断色阶方向：长的那个维度就是渐变方向
if img.width > img.height:
    # 水平色阶，取第一行
    rgb_colours = img_array[0, :, :]
else:
    # 垂直色阶，取第一列
    rgb_colours = img_array[:, 0, :]

3. 处理带透明度的截图 #

如果你的色阶图带有alpha通道（比如PNG格式），可以选择只保留RGB通道，或者把透明度也纳入Colormap：

# 如果图片有4个通道（RGB+Alpha），只保留RGB部分
if img_array.shape[-1] == 4:
    rgb_colours = rgb_colours[:, :3]
    # 要是想保留透明度，直接用rgb_colours就行，Matplotlib支持带alpha的Colormap

小提示 #

• 确保你的截图是纯净的色阶条，没有多余的边框、文字或杂色，不然提取的颜色会有干扰
• 如果觉得生成的Colormap过渡不够平滑，可以把N参数设为256，让颜色更细腻

内容的提问来源于stack exchange，提问作者ダンボー