如何在Matplotlib中绘制颜色条后隐藏负数值区域且不改变配色？

你可以通过保持绘图的原始归一化规则不变，仅调整颜色条的显示范围来实现需求，这样既不会改变现有配色，又能裁剪掉颜色条的负数值部分。具体操作如下：

核心思路 #

• 绘图时继续使用原来的颜色映射（colormap）和归一化（Normalize）配置，确保图像的配色完全不变。
• 针对颜色条，手动调整其显示的轴范围，只展示正数对应的颜色段，同时设置对应的刻度。

代码示例 #

假设你原本的代码是这样的（包含负数范围的配色）：

import matplotlib.pyplot as plt
import numpy as np

# 生成包含正负值的测试数据
data = np.random.randn(10, 10) * 5  # 数据范围约-5到5
# 原始配色和归一化规则
original_cmap = plt.cm.viridis
original_norm = plt.Normalize(vmin=-5, vmax=5)

# 绘制图像
fig, ax = plt.subplots()
im = ax.imshow(data, cmap=original_cmap, norm=original_norm)

现在修改颜色条的显示，只保留正数部分：

# 添加颜色条，extend='min'标记有被截断的最小值（可选）
cb = plt.colorbar(im, extend='min')

# 调整颜色条的显示范围：将轴范围设置为原始归一化后0到最大值对应的区间
cb.ax.set_ylim(original_norm(0), original_norm(5))

# 设置颜色条的刻度为正数范围
cb.set_ticks([0, 1, 2, 3, 4, 5])
cb.set_ticklabels(['0', '1', '2', '3', '4', '5'])

plt.show()

关键说明 #

• original_norm(0)会把数值0转换为原始归一化规则下的0-1区间值，确保颜色条显示的是原配色中对应正数的部分。
• 如果你不需要底部的截断标记，直接去掉extend='min'参数即可。
• 整个过程中，图像的配色完全基于原始的original_norm，不会因为颜色条的调整而改变。

内容的提问来源于stack exchange，提问作者user844843