“输入x和y必须是1D或2D”matplotlib中的错误

Question

我想从一个大文件中绘制一些数据。 数据具有以下形式：

0.025876 139 0 
0.030881 140 0 
0.030982 141 0 
0.035602 142 0 
0.035521 143 0 
0.038479 144 0 
0.040668 145 0 
0.040121 146 0 
0.037953 147 0 
0.039027 148 0 
0.038338 149 0 
0.047557 139 1 
0.045105 140 1 
0.044943 141 1 
0.042370 142 1 
0.042025 143 1 
0.038946 144 1 
0.037953 145 1 
0.033373 146 1 
0.030070 147 1 
0.029118 148 1 
0.025552 149 1 

原则上，每行对应于一个三维点，我会“只是”想绘制从这些点类似于什么我可以用做产生的三维表面在gnuplot中为那些了解它的人绘制splot函数。

你打算在网上找到一个答案，我的问题，我试图与matplolib轮廓功能下列事情：

#!/usr/bin/python 

from numpy import * 
import pylab as p 
import sys 
import mpl_toolkits.mplot3d.axes3d as p3 

s = str(sys.argv[1]) 
f = open(s) 
z,y,x = loadtxt(f, unpack = True) 
f.close 
#x = [1,2,3] 
#y = [1,2,3] 
#z = [1,8,16] 
data = zip(x,y,z) 

#map data on the plane 
X, Y = meshgrid(arange(0, 89, 1), arange(0, 300, 1)) 
Z = zeros((len(X),len(Y)),'Float32') 
for x_,y_,z_ in data: 
    Z[x_, y_] = z_ #this should work, but only because x and y are integers 
      #and arange was done with a step of 1, starting from 0 

fig=p.figure() 
ax = p3.Axes3D(fig) 
ax.contourf(X,Y,Z) 
ax.set_xlabel('X') 
ax.set_ylabel('Y') 
ax.set_zlabel('Z') 
p.show() 

这段代码工作实际罚款的向量x，y和z评论在上面的代码中加上一个hashtag。

但是我知道，我正在尝试使用上面给出的数据，我在matplotlib中得到“输入x和y必须是1D或2D”错误。

我读过这可能与Z不具有与X或Y相同形状的事实有关......但我不确定如何处理此问题。

顺便说一句，你可能已经意识到，我是一个Python新手，我很抱歉如果代码对你们中的一些人来说很丑。

无论如何，任何帮助将非常受欢迎。

谢谢！

法比安斯基

Answer 1

使用scipy.interpolate.griddata：

import io 
import sys 
import numpy as np 
import matplotlib.pyplot as plt 
import mpl_toolkits.mplot3d.axes3d as axes3d 
import scipy.interpolate as interpolate 

content = '''0.025876 139 0 
0.030881 140 0 
0.030982 141 0 
0.035602 142 0 
0.035521 143 0 
0.038479 144 0 
0.040668 145 0 
0.040121 146 0 
0.037953 147 0 
0.039027 148 0 
0.038338 149 0 
0.047557 139 1 
0.045105 140 1 
0.044943 141 1 
0.042370 142 1 
0.042025 143 1 
0.038946 144 1 
0.037953 145 1 
0.033373 146 1 
0.030070 147 1 
0.029118 148 1 
0.025552 149 1''' 

data = np.genfromtxt(io.BytesIO(content), dtype=None, names='x, y, z') 

# Or, to read from a file: 
# data = np.genfromtxt(filename, dtype=None, names='x, y, z') 

x, y, z = data['x'], data['y'], data['z'] 
N = 20 
xi = np.linspace(x.min(), x.max(), N) 
yi = np.linspace(y.min(), y.max(), N) 

X, Y = np.meshgrid(xi, yi) 
Z = interpolate.griddata((x, y), z, (X, Y), method='nearest') 

fig = plt.figure() 
ax = fig.add_subplot(1, 1, 1, projection='3d') 
ax.scatter(data['x'], data['y'], data['z']) 
ax.plot_wireframe(X, Y, Z, rstride=1, cstride=1) 
# ax.plot_surface(X, Y, Z) 
plt.show() 

产生

---

相关链接：

• scipy.interpolate.griddata
• np.genfromtxt
• Axes3D.plot_wireframe
• Axes3D.plot_surface