Python中2d np.array的数组索引？

Question

我遇到了从2d np.array()找到索引的问题。使用If循环，我的算法的计算时间太长。我需要一个numpy函数来从任何数组中找到索引。函数np.where()对我没有帮助。这是一个抽象：

>>> conditions = [[0 0 0 0 4]
[0 0 0 0 3]
[0 0 0 0 2]
..., 
[2 1 3 4 2]
[2 1 3 4 1]
[2 1 3 4 0]]
>>> a = [0, 0, 0, 0, 2]
>>> index = np.where(conditions==a)

如果我这样使用它，虽然我有列和行索引，但我无法解释它们。我需要具体的索引值，例如index = 2。

Answer 1

>>> np.where((conditions ==  a).all(axis=1))[0]
array([2])