python - 使用 matplotlib plot_surface 绘制地形数据

Question

我正在尝试使用 matplotlib 绘制地形高程数据。我构建了一个 nx3 numpy 数组，每行包含我的点的 x、y、z 坐标(它们在 x、y 平面上的网格中有规律地间隔)。我试图用这段代码绘制它:

fig = plt.figure()

ax = fig.gca(projection='3d')

print desiredData[:,0]

surf = ax.plot_surface(desiredData[:,0], desiredData[:,1],
                       desiredData[:,2], rstride =1,
                       cstride = 1, cmap=cm.jet,
                       linewidth = 0, antialiased = False)

plt.show()

但是我收到了这个错误:

Traceback (most recent call last):
   File "gisConvert.py", line 203, in <module>
linewidth = 0, antialiased = False)
File "C:\Python27\lib\site-packages\mpl_toolkits\mplot3d\axes3d.py", line 663,
in plot_surface
rows, cols = Z.shape
ValueError: need more than 1 value to unpack

我做错了什么？

Answer 1

如错误所示，

ValueError: need more than 1 value to unpack

您正在使用一维数组，但 plot_surface 需要 X、Y 和 Z 的二维数组。

这就是您得到 ValueError 的原因。