Matplotlib迭代绘图时3D轴持续缩小的问题求解

我在Tkinter应用中基于指定轴执行迭代3D绘图，无法销毁/重建Figure，只能刷新轴与colorbar。示例代码逻辑为：创建3D轴、绘制3D散点图并添加colorbar，每次迭代移除已有数据和colorbar。

首次迭代显示正常，但后续迭代中轴的显示空间不断缩小，轴与colorbar被限制在上一次迭代的轴空间内。尝试过tight_layout、移除旧数据和colorbar，但没有效果。需求是让轴始终保持首次迭代的完整显示。

示例代码：

import numpy as np
import matplotlib.pyplot as plt
import matplotlib
matplotlib.use("TkAgg")

x = range(10)
y = range(10)
z = range(10)
data = np.random.randn(10, 5)

fig = plt.figure()
ax = fig.add_subplot(projection='3d')
plt.show(block=False)
for i in range(data.shape[-1]):
    if 'colorbar' in fig.get_children()[-1]._label:  # 判断是否为colorbar
        fig.get_children()[-1].remove()  # 移除colorbar
        scat.remove()
        fig.tight_layout()

    scat = ax.scatter3D(x, y, z, c=data[:, i], cmap='hot')
    cb = fig.colorbar(scat)

（注：迭代1为布局正常的3D轴+右侧colorbar；迭代2中3D轴空间被明显压缩，布局紊乱）

问题根源 #

1. 通过fig.get_children()判断并移除colorbar的方式不可靠，fig的子元素顺序会变化，可能导致误判或漏删。
2. 每次添加colorbar时，Matplotlib会自动调整轴的位置以容纳colorbar，但移除colorbar后未恢复轴的原始位置，导致轴空间被反复压缩。
3. tight_layout调用时机错误，且单独调用无法恢复已被修改的轴位置。

解决方案 #

提供两种可靠的修正方案：

方案1：保存并恢复轴的初始位置

通过追踪colorbar对象、清除轴内容、恢复轴初始位置来维持布局：

import numpy as np
import matplotlib.pyplot as plt
import matplotlib
matplotlib.use("TkAgg")

x = range(10)
y = range(10)
z = range(10)
data = np.random.randn(10, 5)

fig = plt.figure()
ax = fig.add_subplot(projection='3d')
# 保存轴的初始位置，用于后续恢复
original_ax_pos = ax.get_position()
cb = None  # 全局追踪colorbar对象
plt.show(block=False)

for i in range(data.shape[-1]):
    # 清除轴上所有绘图元素
    ax.clear()
    # 移除已存在的colorbar并恢复轴位置
    if cb is not None:
        cb.remove()
        ax.set_position(original_ax_pos)
    
    # 绘制新的3D散点图
    scat = ax.scatter3D(x, y, z, c=data[:, i], cmap='hot')
    # 添加新的colorbar
    cb = fig.colorbar(scat)
    # 即时更新画布
    fig.canvas.draw_idle()

方案2：预先分配colorbar的固定空间

使用make_axes_locatable为colorbar预先固定空间，避免轴位置被反复调整：

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import make_axes_locatable
import matplotlib
matplotlib.use("TkAgg")

x = range(10)
y = range(10)
z = range(10)
data = np.random.randn(10, 5)

fig = plt.figure()
ax = fig.add_subplot(projection='3d')
# 创建可分割的轴对象，为colorbar固定右侧空间
divider = make_axes_locatable(ax)
cax = divider.append_axes("right", size="5%", pad=0.1)
cb = None
plt.show(block=False)

for i in range(data.shape[-1]):
    # 清除轴内容
    ax.clear()
    # 移除旧colorbar
    if cb is not None:
        cb.remove()
    
    # 绘制新散点图
    scat = ax.scatter3D(x, y, z, c=data[:, i], cmap='hot')
    # 使用预先分配的cax添加colorbar
    cb = fig.colorbar(scat, cax=cax)
    # 更新画布
    fig.canvas.draw_idle()

方案说明 #

• 方案1通过恢复轴初始位置，确保每次迭代后轴回到首次显示的大小，适合不需要固定colorbar位置的场景。
• 方案2预先为colorbar分配空间，从根源上避免轴位置被修改，布局稳定性更强，推荐使用。

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