如何从轴上获取imshow或冲浪对象？

Question

一旦我在matplotlib中创建了一个子图，请说：

ax.imshow(my_data)

或以3d模式：

ax.plot_surface(x,y,z etc.)

然后将轴对象传递给另一个函数：有没有办法将该轴对象作为变量返回，我们可以操作它？

e.g。

def my_func(ax):
     my_surf = ax.get_surface()
     my_surf.remove()
     ax.plot_surface(new_plot_data...)