如何从 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>]

Answer 2

您可以get_images()在ax.

示例代码：

ax = plt.gca() 
ax.imshow(array) 
ax.get_images() # returns a list of AxesImage objects if any.