python - 从 Skimage 高程图生成 3D 表面图(2D numpy.ndarray)

Question

在the skimage Segmentation tutorial ，绘制了从 sobel 函数生成的高程图的 3D 表面图。

>>> from skimage.filters import sobel
>>> elevation_map = sobel(coins)

问题:elevation_map 似乎是一个 2D numpy.ndarray。我们如何使用它生成显示的 3D map ？

Answer 1

这可能是使用 Paraview/VTK 制作的；

尝试玩以下游戏:

from skimage import data
from skimage.filters import sobel
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
from matplotlib import cm
from scipy.ndimage import zoom
coins = data.coins()
coins = zoom(coins, 10)

elevation_map = sobel(coins)

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

m, n=elevation_map.shape
X, Y = np.meshgrid(np.arange(n), np.arange(m))

ax.plot_surface(X, Y, elevation_map, cmap=cm.viridis, antialiased=False)
ax.axis("off")
ax.set_facecolor('black')
plt.show()