python matplotlib 繪製3D曲面圖

python matplotlib 繪製3D曲面圖

"""
    繪製3D曲面圖
"""
import numpy as np
import matplotlib.pyplot as mp
from mpl_toolkits.mplot3d import Axes3D

# 準備數據
n = 1000
x, y = np.meshgrid(np.linspace(-3, 3, n),
                   np.linspace(-3, 3, n))

z = (1 - x / 2 + x ** 5 + y ** 3) * \
    np.exp(-x ** 2 - y ** 2)

# 繪製圖片
fig = mp.figure("3D Surface", facecolor="lightgray")
mp.title("3D Surface", fontsize=18)

# 設置爲3D圖片類型
ax3d = Axes3D(fig)
# ax3d = mp.gca(projection="3d")    # 同樣可以實現

ax3d.set_xlabel("X")
ax3d.set_ylabel("Y")
ax3d.set_zlabel("Z")
mp.tick_params(labelsize=10)

ax3d.plot_surface(x, y, z, cstride=20, rstride=20, cmap="jet")

mp.show()