Matplotlib 三维绘图

尽管Matplotlib最初只考虑了二维绘图，但后来版本在Matplotlib的二维显示基础上构建了一些三维绘图工具，以提供一套用于三维数据可视化的工具。导入Matplotlib包时，包含了mplot3d工具包。

可以通过向任何常规坐标轴创建程序传递关键字projection=’3d’来创建三维坐标轴。

from mpl_toolkits import mplot3d
import numpy as np
import matplotlib.pyplot as plt
fig = plt.figure()
ax = plt.axes(projection='3d')
z = np.linspace(0, 1, 100)
x = z * np.sin(20 * z)
y = z * np.cos(20 * z)
ax.plot3D(x, y, z, 'gray')
ax.set_title('3D line plot')
plt.show()

我们现在可以绘制各种三维图形。最基本的三维图形是一个 3D线条图 ，由(x, y, z)三元组集合创建。可以使用ax.plot3D函数创建。

Matplotlib 三维绘图

3D散点图 是通过使用 ax.scatter3D 函数生成的。

from mpl_toolkits import mplot3d
import numpy as np
import matplotlib.pyplot as plt
fig = plt.figure()
ax = plt.axes(projection='3d')
z = np.linspace(0, 1, 100)
x = z * np.sin(20 * z)
y = z * np.cos(20 * z)
c = x + y
ax.scatter(x, y, z, c=c)
ax.set_title('3d Scatter plot')
plt.show()

Matplotlib 三维绘图