使用.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

我的问题是，我能得到t2在t中的什么索引，用什么方法

Answer 1

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

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

或者

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

array([0])