如何为matplotlib的imshow设置自定义非线性y轴标签

我明白你遇到的困扰了——imshow 默认会把y轴映射成线性等间距的刻度，哪怕你用了 extent，它也只是把整个y轴范围拉伸到你指定的最大值和最小值，不会自动适配你那组非均匀的 run_numbers。下面给你一个直接可行的解决方案：

核心思路 #

imshow 的y轴本质上对应图像的行索引（从顶部的0到底部的N-1，N是行数）。我们不需要让matplotlib自动生成刻度，而是手动把每个行索引绑定到你自定义的run_numbers值上。

完整代码示例 #

假设你的 profiles 是一个二维数组，行数和 run_numbers 的长度一致：

import matplotlib.pyplot as plt
import numpy as np

# 模拟你的数据（替换成你真实的run_numbers和profiles）
run_numbers = np.array([815676, 815766, 815767, 815800, 815878])
profiles = np.random.rand(len(run_numbers), 500) * 25 - 5  # 匹配vmin=-5, vmax=20的范围

# 绘制图像，保持你原来的参数
plt.imshow(profiles, aspect='auto', vmin=-5, vmax=20)
ax = plt.gca()  # 获取当前坐标轴对象

# 关键步骤：把y轴刻度位置设置为每一行的索引
ax.set_yticks(np.arange(len(run_numbers)))
# 将刻度位置对应的标签替换为你的run_numbers
ax.set_yticklabels(run_numbers)

# 可选：如果希望y轴从上到下是run_numbers从大到小的顺序（和你之前extent的设置一致）
# 因为imshow默认y轴0在顶部，所以如果run_numbers是从小到大排列，反转标签即可
# ax.set_yticklabels(run_numbers[::-1])

# 补充标签和颜色条
plt.xlabel('X Axis')
plt.ylabel('Run Number')
plt.colorbar(label='Profile Value')
plt.show()

为什么之前的方法不行？ #

• 用 extent 时，matplotlib只是把y轴的范围设置为[max(run_numbers), min(run_numbers)]，但它仍然会自动生成等间距的刻度值，这和你非均匀的run_numbers完全不匹配。
• 直接修改 yticklabels 但没设置对应的 yticks 位置的话，matplotlib还是会用原来的线性刻度位置，只是替换了标签文本，导致标签和实际行的对应关系错乱。

额外提示 #

如果你的run_numbers很长（比如有成百上千个值），直接显示所有标签会导致重叠，这时候可以设置刻度的间隔：

# 每隔5行显示一个标签
step = 5
ax.set_yticks(np.arange(0, len(run_numbers), step))
ax.set_yticklabels(run_numbers[::step])

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