Получить экземпляр AxesImage в AxesSubplot

Question

Есть ли какой-нибудь способ получить указатель на экземпляр AxesImage в пределах подзаговора?Например, в общем случае, например:

import matplotlib.pyplot as plt 
import numpy as np

data = np.random.rand(10,10)

fig, ax = plt.subplots(1,1)

ax.imshow(data)

Возвращение объекта AxesImage через что-то вроде ax.get_image.На данный момент я делаю это через:

ls = [type(x) for x in ax.get_children()]
img = ax.get_children()[ls.index(matplotlib.image.AxesImage)]

И мне было интересно, есть ли менее явный способ сделать это.

Answer 1

Как указано DavidG в комментариях, существует метод для получения списка изображений в пределах AxesSubpolot, например:

import matplotlib.pyplot as plt 
import numpy as np

data = np.random.rand(10,10)

fig, ax = plt.subplots(1,1)

ax.imshow(data)

img = ax.get_images()[0] # only one image in this axes instance.