使用Python绘制包含4个数组数据的3D表面图

Question

我将尝试使用具有4列数据的数据绘制3D表面图，即x、y、z、c。下面是所示数据 -

0.452  -172.799  -172.800 0.000000
0.452  -172.799  -158.400 -9.305322
0.452  -172.799  -144.000 -12.062071
0.452  -172.799  -129.600 -9.008698
0.452  -172.799  -115.200 -5.402975
0.452  -172.799  -100.800 -3.957608
0.452  -172.799  -86.400 -5.113347
0.452  -172.799  -72.000 -4.784993
0.452  -172.799  -57.600 -6.659095
0.452  -172.799  -43.200 -4.405937
0.452  -172.799  -28.800 -6.110385
0.452  -172.799  -14.400 -5.896424
0.452  -172.799  -0.001 -4.405937
0.452  -172.799  14.399 -6.489442
0.452  -172.799  28.799 -5.662057
0.452  -172.799  43.199 -7.710180
0.452  -172.799  57.599 -7.710180
0.452  -172.799  71.999 -8.011659
0.452  -172.799  86.399 -6.489442
0.452  -172.799  100.799 -6.817796
0.452  -172.799  115.199 -5.662057
0.452  -172.799  129.599 -5.402975
0.452  -172.799  143.999 -5.662057
0.452  -172.799  158.399 -3.957608
0.452  -172.799  172.799 -2.701488
0.615  -172.799  -172.800 -8.444175
0.615  -172.799  -158.400 -13.183417`

第一、二、三列代表x、y、z，第四列c是颜色条，即根据c给出表面颜色。

目前我尝试了几个简单的代码，但仍然出现错误。我尝试的代码如下：

import pylab as p
import numpy as np 
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import matplotlib.cm as cm
data = np.genfromtxt('test1.pmf')
x=data[:,1]
y=data[:,2]
h=data[:,3]
z=data[:,0]
datamin=min(h)
datamax=max(h)
fig=p.figure()
ax=fig.add_subplot(111, projection='3d')
pmf = ax.plot_surface(x, y, z, facecolors=cm.ocean(h))
p.colorbar(pmf)
p.show()    

我遇到了一个错误，例如：

pmf = ax.plot_surface(x, y, z, facecolors=cm.ocean(h))
File "/usr/lib/pymodules/python2.7/mpl_toolkits/mplot3/axes3d.py",line663,in            
plot_surface rows, cols = Z.shape ValueError: need more than 1 value to unpack

请问我的操作哪里出错了？

非常感谢！

- xyz123

---

您提供的数据中的“x”坐标（第二列）不会改变。您如何期望它成为一个表面呢？ - hitzg

Answer 1

plot_surface函数需要2D数组作为参数。在您的代码片段中，Z变量是一个1D数组，因此Z.shape无法解包成两个元素（行和列）。

我相信，像这样的代码可能是正确的：

x, y = np.meshgrid(x, y)
z = z.reshape(x.shape)

请查看示例

- erthalion

---

我尝试使用meshgrid，但由于我的数据是4D数组，仍然出现错误。我对脚本进行了更改- - xyz123

我尝试了使用meshgrid，由于我的数据是4D数组，我仍然得到了一个错误。我对脚本做了以下更改- X, Y = np.meshgrid(xi, yi) Z = griddata(x, y, z, xi, yi)，但仍然遇到了TypeError: unhashable type: 'numpy.ndarray'的错误提示。 - xyz123

你能否更新你的问题并展示新的代码和新的错误回溯？ - erthalion