Python绘制三维图形

import numpy as np
import matplotlib.pyplot as plt 
import mpl_toolkits.mplot3d

x, y = np.mgrid[-2 : 2 : 20j, -2 : 2 : 20j]
z = 50 * np.sin(x + y)                     # 测试数据
ax = plt.subplot(111, projection = '3d')   # 三维图形
ax.plot_surface(x, y, z, rstride = 2, cstride = 1, cmap = plt.cm.Blues_r)
ax.set_xlabel('x')                         # 设置坐标轴标签
ax.set_xlabel('y')
ax.set_xlabel('z')
plt.show()