Encuentre valores comunes entre dos arrays NumPy – Part 1

En este artículo, vamos a discutir cómo encontrar los valores comunes entre 2 arrays. Para encontrar los valores comunes, podemos usar numpy.intersect1d() , que realizará la operación de intersección y devolverá los valores comunes entre las 2 arrays ordenadas.

Sintaxis: numpy.intersect1d(arr1, arr2, asumir_unique = False, return_indices = False)

Parámetros:
arr1, arr2: [array_like] Arreglos de entrada.
asumir_unique: [bool] Si es verdadero, se supone que las arrays de entrada son únicas, lo que puede acelerar el cálculo. El valor predeterminado es Falso.
return_indices: [bool] Si es True, se devuelven los índices que corresponden a la intersección de las dos arrays. La primera instancia de un valor se usa si hay varios. El valor predeterminado es Falso.

Retorno: [ndarray] Array 1D ordenada de elementos comunes y únicos.

Ejemplo #1: encontrar valores comunes entre arrays 1d

import numpy as np
  
  
# create 2 arrays
a = np.array([2, 4, 7, 1, 4])
b = np.array([7, 2, 9, 0, 5])
  
# Display the arrays
print("Original arrays", a, ' ', b)
  
# use the np.intersect1d method
c = np.intersect1d(a, b)
  
# Display result
print("Common values", c)

Producción:

Original arrays [2 4 7 1 4]   [7 2 9 0 5]
Common values [2 7]

Ejemplo #2: encontrar valores comunes entre arrays n-dimensionales

import numpy as np
  
  
# create 2 arrays
a = np.array([2,4,7,1,4,9]).reshape(3,2)
b = np.array([7,2,9,0,5,3]).reshape(2,3)
  
# Display the arrays
print("Original arrays")
print(a)
print(b)
  
# use the np.intersect1d method
c = np.intersect1d(a,b)
  
# Display result
print("Common values",c)

Producción:

Original arrays
[[2 4]
 [7 1]
 [4 9]]
[[7 2 9]
 [0 5 3]]
Common values [2 7 9]

Nota: Independientemente de las arrays de dimensiones que se pasen, los valores comunes se devolverán de forma aplanada en 1d.