Encuentra la unión de dos arrays NumPy

Para encontrar la unión de dos arrays unidimensionales, podemos usar la función numpy.union1d() de la biblioteca Python Numpy. Devuelve una array única y ordenada con valores que están en cualquiera de las dos arrays de entrada.

Sintaxis:

numpy.union1d(array1, array2)

Nota Las arrays proporcionadas en la entrada se aplanan si no son unidimensionales.

Veamos ejemplos de cómo encontrar la unión de dos arreglos.

Ejemplo 1:

# import libraries
import numpy as np
  
  
arr1 = np.array([10, 20, 30, 40])
print("array1 ", arr1)
  
arr2 = np.array([20, 40, 60, 80])
print("array2 ", arr2)
  
# print union of the two arrays
print("Union of two arrays :", np.union1d(arr1, arr2))

Producción:

Ejemplo 2:
Veamos un ejemplo de encontrar la unión de una array de 2-d y 1-d. Como se discutió anteriormente, si la array pasada como argumentos a la función numpy.union1d es bidimensional, entonces se aplana a una dimensión.

# import libraries
import numpy as np
  
  
# 2-d array
arr1 = np.array([[1, 2, 3], [4, 5, 6]])
print("array1 ")
print(arr1)
  
arr2 = np.array([0, 5, 10])
print("array2 ", arr2)
  
# print union of 2-d array and 1-d array
print("Union of two arrays", np.union1d(arr1, arr2))

Producción:

Ejemplo 3:
si queremos encontrar la unión de más de dos arrays, podemos encontrarla usando la función functools.reduce .

# code to find union of more than two arrays
# import libraries
import numpy as np
from functools import reduce
  
  
array = reduce(np.union1d, ([1, 2, 3], [1, 3, 5],
                            [2, 4, 6], [0, 0, 0]))
print("Union ", array)

Producción: