在二維 Numpy 數組中查找值 x、y、z 的行

Question

我有

import numpy as np
np_array = np.array([[0,4],[0,5],[3,5],[6,8],[9,1],[6,1]])
rows=np.where(np_array[:,0]==6)and np.where(np_array[:,1]==1)
print(np_array[rows])

預期答案

[6,1]

從代碼回答

[[9 1]
 [6 1]]

我想要的是[6,1]所在位置的索引。

我肯定錯過了什么。

Answer 1

沒有where ，只使用索引。

import numpy as np

np_array = np.array([[0, 4], [0, 5], [3, 5], [6, 8], [9, 1], [6, 1]])

rows = (np_array[:, 0] == 6) & (np_array[:, 1] == 1)

print(np_array[rows][0])

您可以使用np.where(rows)獲取索引。