从一个矩形高度数组生成3D表面

Question

我正在尝试在matplotlib中绘制一些HDF数据。使用h5py导入后，数据以数组形式存储，如下所示：

array([[151, 176, 178],
       [121, 137, 130],
       [120, 125, 126])

在这种情况下，x和y值只是数组字段的索引，而z值是特定字段的值。在(x,y,z)形式中，它看起来像：

(1,1,151)
(2,1,176)
(3,1,178)
(1,2,121)
...

等等。

有没有一种简单的方法可以从这种数据中绘制表面图？我知道我可以通过遍历整个数组将其转换为（x，y，z）元组，但也许这并不是必要的？

- Paweł Rumian

Answer 1

如果您想要绘制一个三维曲面图，您首先需要创建meshgrid。您可以尝试使用以下代码：

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import numpy as np

X = np.arange(1, 10)
Y = np.arange(1, 10)
X, Y = np.meshgrid(X, Y)
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)
fig = plt.figure()
ax = fig.gca(projection='3d')
surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap='hot', linewidth=0, antialiased=False)
ax.set_zlim(-1.01, 1.01)

fig.colorbar(surf, shrink=0.5, aspect=5)
plt.show()

这将生成以下图片，

然而，如果只关注 z 值信息，可以直接使用 imshow 函数。在此情况下，z 值将以颜色的形式呈现。您可以通过以下方式实现：

im = plt.imshow(Z, cmap='hot')
plt.colorbar(im, orientation='horizontal')
plt.show()

这将会得到：

- imsc

---

imshow完成了任务，谢谢！由于声望低，我无法投票支持，但我会记住的 :) - Paweł Rumian

它如何回答这个问题？ - Nd.