2つの高次元numpy配列間の相互要素を見つける

Question

私はPythonの初心者です。ここに私の問題があります。私はこれを実行すると

import numpy as np 

def neighbors(indexset, i,j): 
    temp = np.array([[i-1,j],[i+1,j],[i,j-1],[i,j+1]]) 
    for ele in temp: 
     if ele in indexset: 
      print(ele) 

indexset = np.array([[0,1],[1,1],[2,1],[3,1]]) 
neighbors(indexset, 0,0) 

私は

neighbors(indexset, 0,0) 
[1 0] 
[ 0 -1] 
[0 1] 

私が間違って何をやっているを理解していない値を取得しますか？なぜこれは[0,1]しか返しませんか？

Answer 1

ele in tempは、eleがスカラーの場合にのみ適切な結果を得ていると思います。これはnumpy関数__contains__を使用します。これは(a==b).any()に相当します。あなたはPythonのリストの代わりに、numpyの配列でこれを行う場合は、それが動作します：予想通り

def neighbors(indexset, i,j): 
    temp = [[i-1,j],[i+1,j],[i,j-1],[i,j+1]] 
    for ele in temp: 
     if ele in indexset: 
      print(ele) 

indexset = [[0,1],[1,1],[2,1],[3,1]] 
neighbors(indexset, 0,0) 

が[0,1]を印刷します。 indexsetは常にnumpyの配列である場合は、tolistを使用することができます。

サンプル中の上位2次元numpyの配列です

import numpy as np 

def neighbors(indexset, i,j): 
    temp = [[i-1,j],[i+1,j],[i,j-1],[i,j+1]] 
    for ele in temp: 
     if ele in indexset.tolist(): 
      print(ele) 

indexset = np.array([[0,1],[1,1],[2,1],[3,1]]) 
neighbors(indexset, 0,0)