Python 竟能绘制出如此酷炫的三维图
点击上方“高级农民工”，选择“星标”公众号
第一时间速享原创干货
作者 | Jay Alammar
译者 | 高级农民工
转发文章最多的朋友可以免费进入价值99元的《湾区AI精英会》海归圈子
通常我们用 Python 绘制的都是二维平面图，但有时也需要绘制三维场景图，比如像下面这样的：
这些图怎么做出来呢？今天就来分享下如何一步步绘制出三维矢量（SVG）图。
八面体
我们先以下面这个八面体为例。
1 安装相关包
首先安装两个必备包：
import pyrr # NumPy 的 3D 函数库import svgwrite # svg图形处理库# NumPy 的 3D 函数库
import svgwrite # svg图形处理库
2 定义 3D 图生成环境
接下来定义几个类设置好 3 维图基础环境：
viewport ：矩形图范围
camera：包括视图矩阵和投影矩阵
mesh：svg 矢量图所需的网格表面矩阵、着色器和样式字典
3 生成八面体数据
然后生成八面体每个定点的数据：
def octahedron(): """Construct an eight-sided polyhedron""" f = sqrt(2.0) / 2.0 verts = numpy.float32([ ( 0, -1, 0), (-f, 0, f), ( f, 0, f), ( f, 0, -f), (-f, 0, -f), ( 0, 1, 0) ]) triangles = numpy.int32([ (0, 2, 1), (0, 3, 2), (0, 4, 3), (0, 1, 4), (5, 1, 2), (5, 2, 3), (5, 3, 4), (5, 4, 1) ]) return verts[triangles]
"""Construct an eight-sided polyhedron"""
f = sqrt(2.0) / 2.0
verts = numpy.float32([ ( 0, -1, 0), (-f, 0, f), ( f, 0, f), ( f, 0, -f), (-f, 0, -f), ( 0, 1, 0) ])
triangles = numpy.int32([ (0, 2, 1), (0, 3, 2), (0, 4, 3), (0, 1, 4), (5, 1, 2), (5, 2, 3), (5, 3, 4), (5, 4, 1) ])
return verts[triangles]
4 pyrr 包渲染
接下来利用 pyrr 3维库渲染上面的原始数据，最后生成 svg 矢量图。
5 生成 svg
上面最后一步生成 svg 图形，需要调用下面的 Engine 函数，略微有些复杂：
以上代码全部封装到类中调用，就可以生成八面体图形了。
除了八面体还可以生成很多其他图形。
球体和 Klein 瓶
代码如下：
多面球体
代码如下：
发光的球体
代码实现：
还可以绘制这种曲面体
代码实现如下：
以上源代码可以在后台回复：svg 得到。
END.
作者：Jay Alammar
链接：https://jalammar.github.io/visual-numpy/
想脱单，找湾区人工智能
认识帅张，维权不再难
用法律武器，痛击腾讯侵权行为！！！湾区人工智能可以改善知识产权现状吗？
【中美之争】其实已经失去了悬念 (深度好文)|湾区人工智能
【欧洲已死】李开复：欧洲人工智能毫无希望
80后，天才程序员， Facebook 第一任 CTO，看看开挂的人生到底有多变态？
【厂妹进谷歌】从富士康流水线小妹到 Google 工程师，She made it !!！
【刚刚】世界顶级名校：美国斯坦福大学在人工智能领域诞生了一位来自中国的女神
【24万欧】国外人工智能计算机博士自述面试经历
看看你的母校是否开通了人工智能及其相关专业
IT行业高工资的时代已经结束了 | 紫竹张先生
凌晨一点的粤海街道对抗来自美国的力量|湾区人工智能
进入顶层社会的顺序是什么？
程序员的周末是风花雪月还是孤独寂寞？
每天都要刷碗的清贫博士后，生活捉襟见肘，国外华人也不容易
农村的未来，一|深度揭秘俄罗斯远东真相（上）|湾区人工智能
为什么农村出来的大学生大多混得比较差？
【月薪三万】听说深圳老师工资全国最高！！！比德国还高
【幸福的猪】德国难民躺赢的人生：和三个老婆造人就是神圣不可侮辱的职业
长按扫码撩海归
