Поиск взаимных элементов между двумя многомерными массивами 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(). Если вы делаете это с питона списков вместо 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:

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)