今天在进行机器学习画图时遇到画图报错

我的代码是：

plt.figure("Loss Function", facecolor="lightgray")
ax3d = plt.gca(projection="3d")
ax3d.set_xlabel("w0")
ax3d.set_ylabel("w1")
ax3d.set_zlabel("loss")
ax3d.plot_surface(w0_grid, w1_grid, loss, cstride=30, rstride=30, cmap="jet")
plt.show()

运行报错：TypeError: gca() got an unexpected keyword argument 'projection'

查了很多文章，全都是给出这样的修改方法：

#ax = fig.gca(projection='3d')
ax = fig.add_subplot(projection = '3d')

但是我按此修改依然报错不能画图，求教AI，给出另一种答案：
如果你正在使用较新的 Matplotlib 版本，projection 参数可能已经不再被 gca() 支持。在这种情况下，你应该使用 add_subplot() 或者 subplot() 函数来创建带有投影的 Axes 对象。

修改如下：

ax3d = plt.subplot(projection="3d")
ax3d.set_xlabel("w0")
ax3d.set_ylabel("w1")
ax3d.set_zlabel("loss")
ax3d.plot_surface(w0_grid, w1_grid, loss, cstride=30, rstride=30, cmap="jet")
plt.show()

再次运行，正常能画图了！