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

原创已于 2023-03-23 11:22:18 修改 · 6.4k 阅读

20 ·

CC 4.0 BY-SA版权

文章标签：

#pycharm #深度学习 #python #matplotlib

于 2021-10-05 12:43:36 首次发布

PyTorch学习笔记专栏收录该内容

19 篇文章

订阅专栏

本文档详细介绍了如何在Python中利用matplotlib库绘制3D曲面图。首先，讲解了如何在Pycharm中安装matplotlib，通过File——Setting——Python Interpreter进行添加。接着，展示了在matplotlib 3.2.0及以上版本中创建3D图的具体代码，通过`fig.add_subplot(projection='3d')`建立3D坐标系，并使用`ax.plot_surface()`绘制3D曲面图。实例中以W、B、1_sum/3为参数展示了3D图的绘制。最后，提到了在不同版本中导入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