【python系列】python画3d曲

• 2020 年 1 月 7 日
• 筆記

mplot3d example code: surface3d_demo.py

来自：http://matplotlib.org/examples/mplot3d/surface3d_demo.html

from mpl_toolkits.mplot3d import Axes3D  from matplotlib import cm  from matplotlib.ticker import LinearLocator, FormatStrFormatter  import matplotlib.pyplot as plt  import numpy as np    fig = plt.figure()  ax = fig.gca(projection='3d')  X = np.arange(-5, 5, 0.25)  Y = np.arange(-5, 5, 0.25)  X, Y = np.meshgrid(X, Y)  R = np.sqrt(X**2 + Y**2)  Z = np.sin(R)  surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.coolwarm,          linewidth=0, antialiased=False)  ax.set_zlim(-1.01, 1.01)    ax.zaxis.set_major_locator(LinearLocator(10))  ax.zaxis.set_major_formatter(FormatStrFormatter('%.02f'))    fig.colorbar(surf, shrink=0.5, aspect=5)    plt.show()

效果：