如何判断指定Axes对象是否已关联ColorBar？

这里有两种实用的方法可以解决你的问题：

方法一：遍历Figure中的ColorBar对象检查关联 #

通过遍历当前Figure的所有子对象，筛选出ColorBar类型的对象，再检查其关联的绘图对象（mappable）是否属于目标Axes：

import matplotlib.pyplot as plt

def has_associated_colorbar(ax):
    for child in ax.figure.get_children():
        if isinstance(child, plt.Colorbar):
            if child.mappable.axes is ax:
                return True
    return False

使用示例 #

fig, (ax1, ax2) = plt.subplots(1, 2)
# 给ax1添加colorbar
im = ax1.imshow([[0, 1], [2, 3]])
plt.colorbar(im, ax=ax1)

print(has_associated_colorbar(ax1))  # 输出 True
print(has_associated_colorbar(ax2))  # 输出 False

方法二：自定义Axes属性绑定ColorBar（效率更高） #

既然你只需要在目标Axes上创建一次ColorBar，最直接高效的方式是在首次创建时，给Axes添加一个自定义属性来记录已创建的ColorBar，后续直接检查该属性是否存在：

# 假设你要绘制的图像数据为data
im = ax.imshow(data)
# 检查是否已关联colorbar
if not hasattr(ax, '_attached_colorbar'):
    # 创建并绑定到Axes的自定义属性
    ax._attached_colorbar = plt.colorbar(im, ax=ax)

这种方法不需要遍历Figure的子对象，性能更优，尤其适合你需要重复绘制大量对象的场景。

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