查找對應於3D numpy對象中給定z值的x和y值

我在Python（np）中有一個numpy對象，我正在其中繪製pyplot中的圖。這是一個含有x,y和z的3D圖形，其中包含要繪製圖形的值。下面是我的代碼（從here）：查找對應於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()

所以在這裏，我想要得到的x和y，其中z值最接近-30值。我如何獲得x和y的值。

來源

2017-02-07 Mihir Khandekar

如果我理解你的問題是正確的，這可能是有用的http://stackoverflow.com/questions/2566412/find-nearest-value-in-numpy-array –

你可以試試以下（也Find nearest value in numpy array見）

value = 30. 
zid = (np.abs(Z-value)).argmin() 
xval = np.reshape(X, -1)[zid] 
yval = np.reshape(Y, -1)[zid] 
print xval, yval, np.reshape(Z, -1)[zid]

我檢查它的例子（https://gist.github.com/amroamroamro/1db8d69b4b65e8bc66a6）你在問題中提到，它似乎工作。

來源

2017-02-07 09:32:34

您可以usenp.where得到滿足條件-30-eps < z < -30+eps一些公差，eps指數，並利用這些獲得的x和y，例如價值

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

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

我有點不確定究竟是如何爲您的劇情從鏈接的例子值不同，這樣做你的情況（請在問題上MCVE代碼）。這個想法應該沒問題。

來源

2017-02-07 09:29:15

查找對應於3D numpy對象中給定z值的x和y值

回答

相關問題