文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

Python中Matplotlib如何实现3D绘图

2023-06-21 21:32

关注

小编给大家分享一下Python中Matplotlib如何实现3D绘图,相信大部分人都还不怎么了解,因此分享这篇文章给大家参考一下,希望大家阅读完这篇文章后大有收获,下面让我们一起去了解一下吧!

mpl_toolkits 是 Matplotlib 的绘图工具包。

第一个三维绘图程序

下面编写第一个三维绘图程序。

首先创建一个三维绘图区域, plt.axes() 函数提供了一个参数projection,将其参数值设置为 "3d"。如下所示:

#导入三维工具包mplot3dfrom mpl_toolkits import mplot3dimport numpy as npimport matplotlib.pyplot as pltfig = plt.figure()#创建3d绘图区域ax = plt.axes(projection='3d')

有了三维绘图区域,接下来就要构建 3d 图像,如下所示:

#从三个维度构建z = np.linspace(0, 1, 100)x = z * np.sin(20 * z)y = z * np.cos(20 * z)

最后调用 plot3D() 方法绘制 3d 图形,代码如下:

#调用 ax.plot3D创建三维线图ax.plot3D(x, y, z, 'gray')ax.set_title('3D line plot')plt.show()

完整程序如下所示:

from mpl_toolkits import mplot3dimport numpy as npimport matplotlib.pyplot as pltfig = plt.figure()#从三个维度构建z = np.linspace(0, 1, 100)x = z * np.sin(20 * z)y = z * np.cos(20 * z)#调用 ax.plot3D创建三维线图ax.plot3D(x, y, z, 'gray')ax.set_title('3D line plot')plt.show()

输出结果如下所示:

Python中Matplotlib如何实现3D绘图

图1:三维线图(3D Line)

上述代码中的 ax.plot3D() 函数可以绘制各种三维图形,这些三维图都要根据(x,y,z)三元组类来创建。

3D散点图

通过 ax.scatter3D() 函数可以绘制 3D 散点图,示例代码如下:

from mpl_toolkits import mplot3dimport numpy as npimport matplotlib.pyplot as pltfig = plt.figure()#创建绘图区域ax = plt.axes(projection='3d')#构建xyzz = np.linspace(0, 1, 100)x = z * np.sin(20 * z)y = z * np.cos(20 * z)c = x + yax.scatter3D(x, y, z, c=c)ax.set_title('3d Scatter plot')plt.show()

输出结果图:

Python中Matplotlib如何实现3D绘图

图2:Matplotlib 3D绘图

3D等高线图

ax.contour3D() 可以用来创建三维等高线图,该函数要求输入数据均采用二维网格式的矩阵坐标。同时,它可以在每个网格点(x,y)处计算出一个 z 值。

以下示例展示了如何绘制三维正弦等高线图。代码如下:

from mpl_toolkits import mplot3dimport numpy as npimport matplotlib.pyplot as pltdef f(x, y):   return np.sin(np.sqrt(x ** 2 + y ** 2))#构建x、y数据x = np.linspace(-6, 6, 30)y = np.linspace(-6, 6, 30)#将数据网格化处理X, Y = np.meshgrid(x, y)Z = f(X, Y)fig = plt.figure()ax = plt.axes(projection='3d')#50表示在z轴方向等高线的高度层级,binary颜色从白色变成黑色ax.contour3D(X, Y, Z, 50, cmap='binary')ax.set_xlabel('x')ax.set_ylabel('y')ax.set_zlabel('z')ax.set_title('3D contour')plt.show()

输出结果图如下:

Python中Matplotlib如何实现3D绘图

图3:绘制三维轮廓图

3D线框图

线框图同样要采用二维网格形式的数据,与绘制等高线图类似。

线框图可以将数据投影到指定的三维表面上,并输出可视化程度较高的三维效果图。通过 plot_wireframe() 能够绘制 3D 线框图。代码如下:

from mpl_toolkits import mplot3dimport numpy as npimport matplotlib.pyplot as plt#要绘制函数图像def f(x, y):   return np.sin(np.sqrt(x ** 2 + y ** 2))#准备x,y数据x = np.linspace(-6, 6, 30)y = np.linspace(-6, 6, 30)#生成x、y网格化数据X, Y = np.meshgrid(x, y)#准备z值Z = f(X, Y)#绘制图像fig = plt.figure()ax = plt.axes(projection='3d')#调用绘制线框图的函数plot_wireframe()ax.plot_wireframe(X, Y, Z, color='black')ax.set_title('wireframe')plt.show()

输出结果如下:

Python中Matplotlib如何实现3D绘图

图4:Matplotlib绘制线框图

3D曲面图

曲面图表示一个指定的因变量y与两个自变量x和z之间的函数关系。

3D 曲面图是一个三维图形,它非常类似于线框图。不同之处在于,线框图的每个面都由多边形填充而成。Matplotlib 提供的 plot_surface() 函数可以绘制 3D 曲面图,该函数需要接受三个参数值 x,y 和 z 。示例代码如下:

from mpl_toolkits import mplot3dimport numpy as npimport matplotlib.pyplot as plt#求向量积(outer()方法又称外积)x = np.outer(np.linspace(-2, 2, 30), np.ones(30))#矩阵转置y = x.copy().T #数据zz = np.cos(x ** 2 + y ** 2)#绘制曲面图fig = plt.figure()ax = plt.axes(projection='3d')调用plot_surface()函数ax.plot_surface(x, y, z,cmap='viridis', edgecolor='none')ax.set_title('Surface plot')plt.show()

输出结果图:

Python中Matplotlib如何实现3D绘图

图5:Matplotlib绘制曲面图

以上是“Python中Matplotlib如何实现3D绘图”这篇文章的所有内容,感谢各位的阅读!相信大家都有了一定的了解,希望分享的内容对大家有所帮助,如果还想学习更多知识,欢迎关注编程网行业资讯频道!

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     807人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     351人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     314人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     433人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     221人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯