python - 使用 .all() 和 any() 获取搜索数组的索引

Question

我有数组列表，我正在使用 .all() 和 .any() 来检查正在搜索的数组是否在数组列表中。

t = np.array([[1,2,3], [4,5,6] , [7,8,9], [10, 11, 12]])

t2 = np.array([1,2,3])

print((t == t2).all(1))
outpu:> [ True False False False]

这里我可以检查t2是否在t中

print((t == t2).all(1).any())
outpu:> True

我的问题是，我可以使用 .all 和 .any 方法获取 t2 中存在的哪个索引吗？

Answer 1

您正在寻找 np.flatnonzero 或 np.where。

>>> np.flatnonzero((t == t2).all(1))

或者，

>>> np.where((t == t2).all(1))[0]

array([0])