2017-02-07 4 views
1

pyplotにグラフをプロットしているnumpyオブジェクトがPython(np)にあります。 x,yおよびzの3Dグラフは、グラフがプロットされる値を含むリストです。私はzの値が-30に最も近いxy、の値を取得したい、だからここ3D numpyオブジェクトの与えられたz値に対応するxとy値の検索

data = np.c_[x,y,z] 

# regular grid covering the domain of the data 
mn = np.min(data, axis=0) - 0.009 
mx = np.max(data, axis=0) + 0.009 

X,Y = np.meshgrid(np.linspace(mn[0], mx[0]), np.linspace(mn[1], mx[1])) 
XX = X.flatten() 
YY = Y.flatten() 
A = np.c_[np.ones(data.shape[0]), data[:, :2], np.prod(data[:, :2], axis=1), data[:, :2] ** 2] 
C, _, _, _ = scipy.linalg.lstsq(A, data[:, 2]) 

# evaluate it on a grid 
Z = np.dot(np.c_[np.ones(XX.shape), XX, YY, XX * YY, XX ** 2, YY ** 2], C).reshape(X.shape) 
fig = plt.figure() 
ax = fig.gca(projection='3d') 
ax.plot_surface(X, Y, Z, rstride=1, cstride=1, alpha=0.2) 
ax.scatter(data[:, 0], data[:, 1], data[:, 2], c='r', s=50) 
plt.xlabel('X') 
plt.ylabel('Y') 
ax.set_zlabel('Z') 
ax.axis('equal') 
ax.axis('tight') 
plt.show() 

<code>pyplot</code> chart I am getting は、下記(hereから)私のコードです。これらの値を取得するにはxyを入力します。

+0

あなたの質問が正しいと分かっている場合は、こちらが参考になります。http://stackoverflow.com/questions/2566412/find-nearest-value-in-numpy-array –

答えて

1

あなたはusenp.where、いくつかの許容範囲の条件-30-eps < z < -30+epsを満たすインデックスを取得eps、および例えば、xyの値を取得するためにこれらを使用することができます

eps = 10. 
ind = np.where((-30+eps > Z.ravel()) & (Z.ravel() > -30-eps)) 

print("x=",X.ravel()[ind],"y=",Y.ravel()[ind]) 

私は(質問MCVE内のコードを作成してください)あなたのプロットは、リンクから例の値と異なって、あなたのケースでこれを行うには正確にどのように少しわかりませんよ。アイデアはうまくいくはずです。

関連する問題