Indices de l'intersection des lignes de Numpy 2d Tableau

Question

Je veux obtenir les indices des lignes qui se croisent d'une principale numpy tableau 2D A, avec un autre B.

A = array([[1,2], 
      [1,3], 
      [2,3], 
      [2,4], 
      [2,5], 
      [3,4] 
      [4,5]]) 

B = array([[1,2], 
      [3,2], 
      [2,4]]) 

result=[0, -2, 3] 
##Note that the intercept 3,2 must assign (-) because it is the opposite 

Lorsque cela devrait revenir [0, -2 , 3] basé sur les indices du tableau A.

Merci!

Answer 1

Vous pouvez référencer le code.

import numpy as np 
A = np.array([[1,2], 
      [1,3], 
      [2,3], 
      [2,4], 
      [2,5], 
      [3,4], 
      [4,5]]) 

B = np.array([[1,2], 
      [3,2], 
      [2,4]]) 

result=[] 

for i in range(0, len(A)): 
    for j in range(0, len(B)): 
     if A[i].tolist() == B[j].tolist(): 
      result.append(i) 
     if A[i].tolist()[::-1] == B[j].tolist(): 
      result.append(-i) 
print(result) 

La sortie est:

[0, -2, 3] 

Answer 2

Le paquet numpy_indexed (disclaimer: Je suis son auteur) a des fonctionnalités pour résoudre efficacement ces problèmes.

import numpy_indexed as npi 
A = np.sort(A, axis=1) 
B = np.sort(B, axis=1) 
result = npi.indices(A, B) 
result *= (A[:, 0] == B[:, 0]) * 2 - 1