Matplotlib是一个Python库,用于创建各种类型的图表,包括3D图。要绘制3D图,需要使用 matplotlib.pyplot 和 mpl_toolkits.mplot3d 两个模块。
首先,导入这两个模块:
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
然后,在绘图之前,创建一个3D坐标系,并将其设置为当前图表:
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
最后,使用ax.plot,ax.scatter等绘制3D图
举个例子,绘制一个3D散点图
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
x = np.random.normal(0, 1, 100)
y = np.random.normal(0, 1, 100)
z = np.random.normal(0, 1, 100)
ax.scatter(x, y, z)
plt.show()
这样就可以画出一个3D散点图了。