在numpy数组中使用y查找特定的x值

时间:2018-07-16 08:30:47

标签: python numpy

我在python中有一个numpy数组[x,y],如下所示:

myarr=np.array([[6,15],[5,10],[7,7],[11,7],[15,10],[13,15]])

print(np.where(myarr==15))

保存x,y坐标

例如,我需要找到x值,其中y为15,在这种情况下为6和13。

我尝试使用np.where解决此问题,但找不到答案。 我已经尝试过了

print(np.where(myarr[1]==15))

但是它给了我空值

1 个答案:

答案 0 :(得分:1)

myarr [1]表示行索引1行。行索引1行中没有15。

找到所有行和列索引1。

print ( np.where(myarr[:,1]==15) )

和x值是

print ( myarr[ np.where(myarr[:,1]==15), 0 ][0] )