def draw3D(X, Y, Z, angle): fig = plt.figure(figsize=(15,7)) ax = Axes3D(fig) ax.view_init(angle[0], angle[1]) ax.plot_surface(X,Y,Z,rstride=1, cstride=1, cmap='rainbow') plt.imshow