使用.all（）和any（）获取搜索数组的索引

Question

我有数组列表，我使用.all（）和.any（）来检查数组im搜索是否在数组列表中。

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

我的问题是，我可以得到.t和.any方法存在的t2中的索引吗？

Answer 1

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

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

或者，

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

array([0])