创建三维轴

默认情况下,Matplotlib 轴是二维的。为了创建三维图,我们需要从 mplot3d 工具包导入 Axes3D 类,这将为轴启用一种新的投影,即'3d'

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

除了二维图的直接概括(例如线图散点图条形图等高线图 )之外,还有几种表面绘图方法 ,例如 ax.plot_surface

# generate example data
import numpy as np
x,y = np.meshgrid(np.linspace(-1,1,15),np.linspace(-1,1,15))
z = np.cos(x*np.pi)*np.sin(y*np.pi)

# actual plotting example
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
# rstride and cstride are row and column stride (step size)
ax.plot_surface(x,y,z,rstride=1,cstride=1,cmap='hot')
ax.set_xlabel(r'$x$')
ax.set_ylabel(r'$y$')
ax.set_zlabel(r'$\cos(\pi x) \sin(\pi y)$')
plt.show()

StackOverflow 文档