python - 您将如何在 Python 中绘制此 3D 可视化图？

Question

我想拍摄一张图像，并将图像的维度表示为空间坐标，将像素值表示为由高度表示的第 3 维。这是一个 example我想做的事。

你会如何使用 matplotlib 或 plotly 做到这一点？

Answer 1

假设图像是 HWC 格式，是灰度(即 C=1)并且是一个 numpy 数组，类似于:

import matplotlib.pyplot as plt
import numpy as np

img = np.random.uniform(size=(10,10,1))

X = np.arange(0, img.shape[1])
Y = np.arange(0, img.shape[0])
X, Y = np.meshgrid(X, Y)
Z = img.squeeze()

# Plot the surface.
fig = plt.figure()
ax = fig.gca(projection='3d')
surf = ax.plot_surface(X, Y, Z)

灵感来自:this .希望这对您有所帮助!