【Python】Axes3D绘图在jupyter里显示不出来的解决办法

一、问题描述

在进行多元线性回归模型的可视化展示的时候，需要用到Axes3D库进行绘图，但是出现了问题，代码没有报错，但是图像显示不出来，如下图所示：

二、问题分析

Axes3D(fig)

将自身添加到图像中，自3.4版本之后被弃用。

三、问题解决

按照提示，使用如下代码替换 ax = Axes3D(fig) 即可：

ax = Axes3D(fig,auto_add_to_figure=False)
fig.add_axes(ax)

结果顺利展现！

四、继续优化

发现坐标轴的负数没办法显示，新加两行代码：

import matplotlib
matplotlib.rcParams['axes.unicode_minus']=False

z轴的标签没有显示出来，是尺寸不够，需要加宽。

最终代码如下：

from mpl_toolkits.mplot3d import Axes3D
import matplotlib
matplotlib.rcParams['axes.unicode_minus'] = False
x1 = np.linspace(X[:, 1].min(), X[:, 1].max(), 100)
x2 = np.linspace(X[:, 2].min(), X[:, 2].max(), 100)
x1, x2 = np.meshgrid(x1, x2)
f = final_theta[0, 0] + final_theta[0, 1] * x1 + final_theta[0, 2] * x2
fig = plt.figure(figsize = (10, 12))
ax = Axes3D(fig, auto_add_to_figure = False)
fig.add_axes(ax)
ax.plot_surface(x1, x2, f, rstride = 1, cstride = 1, label = 'prediction')
ax.scatter(X[:, 1], X[:, 2], y, c='black')
ax.set_zlabel('y')
ax.set_ylabel('X1')
ax.set_xlabel('X2')
plt.show()