如何更简便地获取AxesSubplot中的AxesImage实例？

当然有更优雅简洁的方式来获取AxesImage实例，不用手动遍历get_children()做类型判断！这里有两种推荐的方法：

方法一：直接保存imshow()的返回值 #

这是最直接高效的方式——因为ax.imshow()本身就会返回创建的AxesImage对象，你只需要在调用时把它赋值给变量即可：

import matplotlib.pyplot as plt
import numpy as np

data = np.random.rand(10,10)
fig, ax = plt.subplots(1,1)
# 直接获取AxesImage实例
img = ax.imshow(data)

这种方法不仅简洁，还能避免后续查找的麻烦，是最推荐的做法。

方法二：利用Axes的images属性 #

如果是事后需要从已有的Axes对象中获取（比如不是你自己创建的轴，或者已经运行过绘图代码），可以直接访问ax.images属性——这是Matplotlib专门为存储AxesImage实例提供的列表：

# 假设ax已经存在，且包含一个AxesImage
img = ax.images[0]

# 如果有多个图像，可以遍历筛选
for image in ax.images:
    # ax.images里的元素本身就是AxesImage类型，可直接处理
    print(image)

ax.images只会包含当前轴上所有的AxesImage对象，不用像你之前那样遍历所有子元素再判断类型，可读性和效率都更高。

对比你原来的实现，这两种方法都更简洁可靠，避免了手动查找类型索引可能出现的错误（比如轴子元素顺序变化导致索引不对的情况）。

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