PyTorch深度学习——PyCharm中使用matplotlib画3D曲面图

本篇仅针对使用matplotlib画3D曲面图，其他类型的3D图可根据本实例举一反三进行绘制。

具体步骤：

1、Pycharm如果没有matplotlib，第一部需要先安装matplotlib  

• 点击File——Setting

• 选择 Project:deeplearning(项目名字) ——>Python Interpreter ,点击图中所示“加号”

 

•  点击”加号“之后在跳出的页面中，搜索matplotlib，选择matplotlib，点击Install Package，而后进行安装。

•  安装成功会出现的内容如图：

2、安装matplotlib成功之后，就可以在项目中成功引用和使用matplotlib， 首先引入matplotlib.pyplot，而后将projection="3d" 关键字参数传递给来创建的Figure.add_subplot，具体代码如下：以下代码画的是 以W、B、1_sum/3为参数的3D曲面图。

import matplotlib.pyplot as plt

# 引入matplotlib 3D画图
fig = plt.figure()
ax = fig.add_subplot(projection='3d')

# 引入matplotlib 中的3D曲面图
ax.plot_surface(W, B, l_sum / 3)   #将W,B,1_sum/3 三个参数传入
plt.show()

 注：在 3.2.0 版更改：在 Matplotlib 3.2.0 之前，有必要显式导入 mpl_toolkits.mplot3d模块以使 '3d' 投影到 Figure.add_subplot.

plot_surface()中更多参数的取值、规格以及更多类型的3D图，见官网：The mplot3d Toolkit — Matplotlib 3.4.3 documentation