¿Cómo construir una array de todas las combinaciones de dos arrays NumPy?

A veces necesitamos encontrar la combinación de elementos de dos o más arreglos. Numpy tiene una función para calcular la combinación de 2 o más arrays Numpy llamadas » numpy.meshgrid() «.

Sintaxis:

numpy.meshgrid(*xi, copy=True, sparse=False, indexing='xy')

Ejemplo 1: cálculo de combinaciones de elementos de dos arrays NumPy

# importing Numpy package
import numpy as np
  
# creating 2 numpy arrays
array_1 = np.array([1, 2])
array_2 = np.array([4, 6])
  
print("Array-1")
print(array_1)
  
print("\nArray-2")
print(array_2)
  
# combination of elements of array_1 and array_2
# using numpy.meshgrid().T.reshape()
comb_array = np.array(np.meshgrid(array_1, array_2)).T.reshape(-1, 2)
  
print("\nCombine array:")
print(comb_array)

Producción:

En el ejemplo anterior, combinamos elementos de ‘ array_1 ‘ y ‘ array_2 ‘ usando numpy.meshgrid().T.reshape() 

Ejemplo 2: cálculo de combinaciones de elementos de tres arrays NumPy

# importing Numpy package
import numpy as np
  
# creating 3 numpy arrays
array_1 = np.array([1, 2, 3])
array_2 = np.array([4, 6, 4])
array_3 = np.array([3, 6])
  
print("Array-1")
print(array_1)
  
print("Array-2")
print(array_2)
  
print("Array-3")
print(array_3)
  
  
# combination of elements of array_1,
# array_2 and array_3 using 
# numpy.meshgrid().T.reshape()
comb_array = np.array(
  np.meshgrid(array_1, array_2, array_3)).T.reshape(-1, 3)
  
print("\nCombine array:")
print(comb_array)

Producción:

En el ejemplo anterior, combinamos elementos de ‘ array_1 ‘, ‘ array_2 ‘ y ‘ array_3 ‘ usando numpy.meshgrid().T.reshape()

Ejemplo 3: cálculo de combinaciones de elementos de cuatro arrays NumPy

# importing Numpy package
import numpy as np
  
# creating 4 numpy arrays
array_1 = np.array([50, 21])
array_2 = np.array([4, 4])
array_3 = np.array([1, 10])
array_4 = np.array([7, 14])
  
  
print("Array-1")
print(array_1)
  
print("Array-2")
print(array_2)
  
print("Array-3")
print(array_3)
  
print("Array-4")
print(array_4)
  
  
# combination of elements of array_1, 
# array_2, array_3 and array_4
# using numpy.meshgrid().T.reshape()
comb_array = np.array(np.meshgrid(
    array_1, array_2, array_3, array_4)).T.reshape(-1, 4)
  
print("\nCombine array:")
print(comb_array)

Producción:

En el ejemplo anterior, combinamos elementos de ‘ array_1 ‘, ‘ array_2 ‘, ‘ array_3 ‘ y ‘ array_4 ‘ usando numpy.meshgrid().T.reshape()