文章作者：Tyan
博客：noahsnail.com ?|? CSDN ?|? 簡書
本文主要使用matplotlib進行3D圖像的繪制仰冠。
- Demo
import matplotlib.pyplot as plt
import numpy as np
from mpl_toolkits.mplot3d import Axes3D
# 定義figure
fig = plt.figure()
# 將figure變?yōu)?d
ax = Axes3D(fig)
# 數(shù)據(jù)數(shù)目
n = 256
# 定義x, y
x = np.arange(-4, 4, 0.25)
y = np.arange(-4, 4, 0.25)
# 生成網(wǎng)格數(shù)據(jù)
X, Y = np.meshgrid(x, y)
# 計算每個點對的長度
R = np.sqrt(X ** 2 + Y ** 2)
# 計算Z軸的高度
Z = np.sin(R)
# 繪制3D曲面
ax.plot_surface(X, Y, Z, rstride = 1, cstride = 1, cmap = plt.get_cmap('rainbow'))
# 繪制從3D曲面到底部的投影
ax.contour(X, Y, Z, zdim = 'z', offset = -2, cmap = 'rainbow')
# 設置z軸的維度
ax.set_zlim(-2, 2)
plt.show()
- 結(jié)果
圖像
