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

Question

我在 python 中有一个 numpy array[x,y] ，如下所示：

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

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

它拥有 x,y 坐标

例如，我需要找到 y 为 15 的 x 值，在本例中为 6 和 13。

我试图解决这个问题，np.where但我找不到答案。我也已经尝试过了

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

但它给了我空值

标签： pythonnumpy

Answer 1

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

在所有行和列索引 1 中查找。

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

和 x 值是

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