PyqtGraph ColorBarItem如何调整缩放比例与摆放位置

基于ImageItem与ColorBarItem组件构建热力图时，默认生成的色条（ColorBar）尺寸过小，固定显示在图像左上角。需要将色条调整为与热力图等高，同时放置在图像右侧，需明确修改ColorBarItem的scale（缩放）与translate（平移）属性的实现方式。

原有实现代码 #

def init_plot(self, max, min):
    self.win = pg.GraphicsWindow()
    view = self.win.addViewBox()
    view.setAspectLocked(True)
    view.invertY()
    self.pressure_map = PressureMap("foot_preprocess.png", mode=COMPLETE) # 用于更新图像颜色的自定义类，运行正常
    self.image = pg.ImageItem(axisOrder='row-major')
    self.update() # 用于更新图像色彩，运行正常
    view.addItem(self.image)
    cm = pg.colormap.get('CET-L9')
    lut = cm.getLookupTable(0.0, 1.0)
    self.image.setLookupTable(lut)
    self.image.setLevels([min, max])
    bar = pg.ColorBarItem(values=(min, max), cmap=cm)
    view.addItem(bar)

不推荐手动硬写scale和translate的固定值，这类硬编码参数会在窗口缩放、图像尺寸更新时出现错位，优先使用组件自带的适配方法，代码改动量最小且适配效果稳定。

推荐方案：自动适配（无需手动计算缩放平移） #

ColorBarItem原生提供了setImageItem方法，绑定目标ImageItem后会自动匹配图像高度、自动定位到图像右侧，仅需要在初始化色条时指定合适的宽度即可：

def init_plot(self, max_val, min_val):
    self.win = pg.GraphicsWindow()
    view = self.win.addViewBox()
    view.setAspectLocked(True)
    view.invertY()
    self.pressure_map = PressureMap("foot_preprocess.png", mode=COMPLETE)
    self.image = pg.ImageItem(axisOrder='row-major')
    self.update()
    view.addItem(self.image)
    cm = pg.colormap.get('CET-L9')
    lut = cm.getLookupTable(0.0, 1.0)
    self.image.setLookupTable(lut)
    self.image.setLevels([min_val, max_val])
    # 初始化色条时指定固定宽度，可根据显示需求调整数值
    bar = pg.ColorBarItem(
        values=(min_val, max_val),
        cmap=cm,
        width=20,
        interactive=False
    )
    # 绑定图像，自动完成等高缩放、右侧定位
    bar.setImageItem(self.image, insert_in=view)

注：原代码中使用max、min作为参数名会覆盖Python内置函数，建议修改为max_val、min_val避免潜在问题。

手动设置scale与translate的参考逻辑 #

如果有特殊场景必须手动调整变换属性，可按照以下逻辑计算参数：

• 先获取图像的显示边界：调用self.image.boundingRect()拿到图像的宽高、坐标范围
• 设置缩放：x方向缩放值为目标色条宽度（比如20像素），y方向缩放值等于图像高度，保证色条和图像等高
• 设置平移：x方向平移量为「图像右边界坐标 + 预留间距」，y方向平移量对齐图像顶部即可

参考代码如下：

img_rect = self.image.boundingRect()
bar_width = 20
gap = 5 # 色条和图像的间距，单位像素
# 先重置色条变换，避免多次设置变换导致偏移累积
bar.resetTransform()
# 设置缩放
bar.scale(bar_width, img_rect.height())
# 设置平移
bar.translate(
    (img_rect.right() + gap) / bar_width,
    img_rect.top() / img_rect.height()
)

手动设置的参数不会随视图、图像尺寸变化自动更新，每次视图调整后需要重新计算赋值。

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