numpy where：如何通過匹配一維數組在 3D 數組中找到二維數組的索引？

Question

a = np.array([[[1,2,3], (1,1,0,0,8)], [[1,2,3], (1,1,2,0,8)], [[1,2,3], (4,1,0,0,8)]])
where = np.where(a[:,1] == (1,1,0,0,8))
print(where)

輸出：

(array([], dtype=int64),)

我想要它 output 0 ，出現(1,1,0,0,8)的行的索引

Answer 1

NumPy 通常將元組解釋為數組或數組的子集而不是元素，但您可以通過以下方式繞過它：

el = np.array([None], dtype = object)
el[0] = (1,1,0,0,8)

where = np.where( a[:,1] == el ) # outputs a tuple: (array([0], dtype=int64),)