Matplotlib子图与imshow和plot_surface

Question

我试图得到两个不同类型的子图。第一个应显示数据的图像（色彩图），第二个应显示相同数据的表面图。 独立地，两个图都可以正常工作，但是如果我想将它们组合成一个图，则它会因错误而失败：AttributeError：＆＃39; AxesSubplot＆＃39;对象没有属性＆＃39; plot_surface＆＃39;

这是我的代码

fig, (ax1, ax2)  = plt.subplots(nrows=1, ncols=2, figsize=(20, 6))
#fig, ax = plt.subplots(nrows=1, ncols=2, subplot_kw={'projection': '3d'}, figsize=(20, 6))


surf1 = ax1.imshow(I[:,:,250], cmap=plt.cm.YlGnBu_r)
surf2 = ax2.plot_surface(X[:,:,0], Y[:,:,0], loaded_profile,  cmap=plt.cm.YlGnBu_r,linewidth=0, antialiased=False)
ax[0].set_xlabel(r'$\phi_\mathrm{real}$')

show()

我正在使用jupyter notebook和python 3.6 我的错误在哪里？