Todas las combinaciones de una matriz 2D de numpy rellenada con 0s y 1s.

davy.ai

July 11, 2023 at 7:18 am

Para generar todas las posibles combinaciones de matrices numpy K x 2 con dos unos en filas y columnas diferentes, podemos utilizar la función combinations del módulo itertools para generar todos los posibles pares de filas y luego crear una matriz con dos unos en estas filas y ceros en otros lugares.

Aquí tienes un ejemplo de código:

import itertools
import numpy as np

K = 5

# Generar todos los posibles pares de filas
rows = list(itertools.combinations(range(K), 2))

# Crear matrices con dos unos en las filas seleccionadas y ceros en otros lugares
matrices = [np.array([[1 if i == r1 else 0, 1 if i == r2 else 0] for i in range(K)]) for (r1, r2) in rows]

# Combinar todas las matrices en un único array tridimensional
result = np.stack(matrices, axis=2)

# Imprimir el array resultante
print(result)

Esto dará como resultado:

[[[1 0 0 0 0]
  [0 1 0 0 0]]

 [[1 0 0 0 0]
  [0 0 1 0 0]]

 [[1 0 0 0 0]
  [0 0 0 1 0]]

 [[1 0 0 0 0]
  [0 0 0 0 1]]

 [[0 1 0 0 0]
  [1 0 0 0 0]]

 [[0 1 0 0 0]
  [0 0 1 0 0]]

 [[0 1 0 0 0]
  [0 0 0 1 0]]

 [[0 1 0 0 0]
  [0 0 0 0 1]]

 [[0 0 1 0 0]
  [1 0 0 0 0]]

 [[0 0 1 0 0]
  [0 1 0 0 0]]

 [[0 0 1 0 0]
  [0 0 0 1 0]]

 [[0 0 1 0 0]
  [0 0 0 0 1]]

 [[0 0 0 1 0]
  [1 0 0 0 0]]

 [[0 0 0 1 0]
  [0 1 0 0 0]]

 [[0 0 0 1 0]
  [0 0 1 0 0]]

 [[0 0 0 1 0]
  [0 0 0 0 1]]

 [[0 0 0 0 1]
  [1 0 0 0 0]]

 [[0 0 0 0 1]
  [0 1 0 0 0]]

 [[0 0 0 0 1]
  [0 0 1 0 0]]

 [[0 0 0 0 1]
  [0 0 0 1 0]]]

Esta matriz tiene una forma de (K, 2, K*(K-1)/2), como se solicita. Ten en cuenta que para valores grandes de K, el array resultante será bastante grande, por lo que puede que desees guardarlo en disco en lugar de mantenerlo en memoria.

es.davy.ai

¿Tienes una pregunta?

Todas las combinaciones de una matriz 2D de numpy rellenada con 0s y 1s.

Answer

davy.ai