アレイのインデックスを配列の値で見つける方法は？

Question

数値で配列のインデックスを検索しようとしています。しかし、値は配列でもあります。つまり、多次元配列です。例えば

：

a = [ 
[[1, 0], [0, 2], [3, 3]], 
[[1, 0], [1, 3], [1, 0]], 
[[4, 0], [2, 3], [3, 0]] 
] 

私は1ある[[1, 0], [1, 3], [1, 0]]のインデックスを、見つけたいです。基本的には、私が持っている配列パターンと一致する配列内の要素を探したいと思います。

numpyを使用してこれを行うにはどうすればよいですか？

Answer 1

放送比較と併せて使用np.flatnonzero：他の回答から

a 

array([[[1, 0], 
     [0, 2], 
     [3, 3]], 

     [[1, 0], 
     [1, 3], 
     [1, 0]], 

     [[4, 0], 
     [2, 3], 
     [3, 0]]]) 

np.flatnonzero((a == [[1, 0], [1, 3], [1, 0]]).all(1).all(1)) 
array([1]) 

借入、あなたはallに複数の軸を渡すことができます。

は

np.flatnonzero((a == [[1, 0], [1, 3], [1, 0]]).all((1, 2))) 
array([1]) 

Answer 2

を使用できnp.allとnp.where：

np.where(np.all(a==template, axis=(1,2)))[0][0] 
# 1