Matplotlib 3D 表面颜色图 - AttributeError: 'Axes3DSubplot' 对象没有属性 'cmap'

Question

我正在尝试使用下面的代码绘制 3d 表面颜色图。 它给出了那个错误。 我究竟做错了什么？ 我正在使用 Matplotlib 3.3.2

当我删除 cmap 时，此代码工作正常。

x=np.linspace(-5,6, num=11)
y=np.linspace(-5,6, num=11)

x, y = np.meshgrid(x, y)

z = g1x

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d', cmap=cm.coolwarm)
ax.plot_surface(x,y,z)
plt.show()