解决ax = fig.gca(projection=’3d’)报错的方法

最近在学习tensorflow，其中有使用matplotlib画一个3d图像的练习。跑代码时遇到以下问题

TypeError: FigureBase.gca() got an unexpected keyword argument 'projection

我发现是Axes3D函数没有用到（图中第5行），通过查询并实验网上的一些解决方法后，现记录一下

解决方法：

ax = fig.gca(projection='3d')                                #原来的代码
ax = fig.add_axes(Axes3D(fig))                            #改正后的代码

实际效果如图所示：