matplotlib - 如何从 matplotlib 获取 AxesImages

Question

所有，我使用这样的代码来绘制图像

import matplotlib.pyplot as plt

plt.imshow(array,cmap='jet')
plt.show()

但是，现在我想获取 im 的句柄( im=plt.imshow(array,cmap='jet') )
如何获取 im的句柄如果我在第二步中忽略句柄。

Answer 1

您可以检索轴上存在的所有子项并过滤对象类型。

喜欢:

ax = plt.gca()
imgs = [obj for obj in ax.get_children() if isinstance(obj, mpl.image.AxesImage)]

如果轴上只有一个 AxesImage，它会返回一个包含一个对象的列表，例如:

[<matplotlib.image.AxesImage at 0x7be6400>]