Matplotlib交互式3D网格中静态轴颜色条交互异常问题

问题现象 #

在Matplotlib交互式3D绘图中，为Colorbar单独设置轴以避免随3D网格旋转，但交互时出现异常：

• 初始状态下绘图和Colorbar显示正常
• 第一次操作滑块/按钮时，Colorbar消失
• 第二次交互后，绘图数据也随之消失
• 若不使用cb.remove()，每次交互都会重复生成Colorbar，导致堆叠混乱

问题根源 #

1. 重复绑定事件：在update方法中每次都重新绑定按钮的点击事件，导致事件触发逻辑混乱，后续交互会执行多次无效操作
2. Colorbar移除逻辑不当：cb.remove()会彻底移除Colorbar及其关联的部分资源，后续重新创建时无法正确关联到预设的cax轴

修复后的代码 #

import matplotlib.pyplot as plt
from matplotlib.widgets import Slider, Button
import matplotlib.cm as cm
from matplotlib.colors import Normalize
import numpy as np
%matplotlib notebook

class InteractivePlot:
    def __init__(self):
        self.fig = plt.figure(layout='constrained')
        self.ax = self.fig.add_subplot(111, projection='3d')
        self.cmap = cm.get_cmap('Blues')
        
        # 初始化数据和ScalarMappable
        self.x = np.random.randint(1, 3, size=(4, 4)) 
        self.normalizer = Normalize(np.min(self.x), np.max(self.x))
        self.im = cm.ScalarMappable(norm=self.normalizer, cmap=self.cmap)
        
        # 创建Colorbar的轴并生成Colorbar
        self.cax_pos = [0.93, 0.15, 0.02, 0.7]
        self.cax = self.fig.add_axes(self.cax_pos)
        self.cb = self.fig.colorbar(self.im, cax=self.cax)

        # 初始化网格和绘图
        self.X, self.Y = np.meshgrid(np.linspace(0,1,self.x.shape[1]), np.linspace(0,1,self.x.shape[0]))
        self.contour = self.ax.contourf(self.x, self.X, self.Y, offset=0, cmap=self.cmap)
        
        # 创建按钮并绑定事件（仅绑定一次）
        self.button1 = Button(plt.axes([0.85, 0.01, 0.05, 0.05]), '-', hovercolor='white')
        self.button2 = Button(plt.axes([0.9, 0.01, 0.05, 0.05]), '+', hovercolor='white')
        self.button1.on_clicked(self.decrement_slider)
        self.button2.on_clicked(self.increment_slider)
        
        # 创建滑块并绑定更新事件
        slider_ax = plt.axes([0.15, 0.01, 0.6, 0.03])
        self.slider = Slider(slider_ax, 'Upper lim', 1, 10, valinit=3, valstep=1)
        self.slider.on_changed(self.update)

    def increment_slider(self, event):
        self.slider.set_val(self.slider.val + self.slider.valstep)

    def decrement_slider(self, event):
        self.slider.set_val(self.slider.val - self.slider.valstep)

    def update(self, val):
        # 清除旧的绘图内容
        self.ax.clear()
        
        # 生成新数据
        self.x = np.random.randint(1, self.slider.val, size=(4, 4)) 
        # 更新归一化参数
        self.normalizer.vmin = np.min(self.x)
        self.normalizer.vmax = np.max(self.x)
        self.im.set_norm(self.normalizer)
        
        # 重新绘制contourf
        self.contour = self.ax.contourf(self.x, self.X, self.Y, offset=0, zdirs='z', cmap=self.cmap)
        # 更新Colorbar
        self.cb.update_normal(self.im)
        
        # 刷新画布
        self.fig.canvas.draw_idle()

# 启动交互绘图
plot = InteractivePlot()
plt.show()

关键修改说明 #

1. 事件绑定优化：将按钮的增减逻辑单独提取为方法，并在__init__中仅绑定一次，避免重复绑定导致的逻辑混乱
2. Colorbar更新方式：不再移除和重建Colorbar，而是通过更新ScalarMappable的归一化参数，调用cb.update_normal(self.im)来更新Colorbar，避免资源销毁和重建的问题
3. 代码结构简化：清理update方法中的冗余代码，让逻辑更清晰，减少不必要的对象创建和销毁

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