Генерация комбинации строк матрицы - PullRequest
Новая
       16

Генерация комбинации строк матрицы

0 голосов
/ 09 октября 2018

Например, у меня матрица с 64 строками.Я хочу, чтобы все комбинации выглядели так, как будто мы можем взять 1 элемент из строк 0–3, 1 элемент из 4–7 и т. Д. Поэтому мне нужно 16 строк из матрицы.Моя идея сделать 16 матриц с 4 рядами и попытаться выбрать один из каждого.Но мой код не работает.Задача - получить все комбинации строк, где мы берем одну из всех групп по 4 строки.Мой код: 

matrix = np.random.randint(0, 1, (64,64))  # random 64X64 matrix with zeros and ones
matrix1 = [matrix[4*k:4*k+4:1] for k in range(16)]  # getting 16 matrix like 0..3,4...7 etc
for checker in itertools.combinations(itertools.chain([[matrix1[i] for i in range(16)]])):
    print(str(checker))  # do smth

1 Ответ

0 голосов
/ 09 октября 2018

Для чего-то большего, чем у вас, очень сложно напечатать результаты, поэтому я приведу рабочий пример для чего-то меньшего - 

import numpy as np
from itertools import permutations
matrix = np.random.randint(0, 5, (4,2))
print(matrix)

Вывод: 

[[4 3]
 [3 0]
 [1 3]
 [3 3]]

Получение перестановок: 

k = permutations(matrix, 2)
for sub_matrix in k:
    print(sub_matrix)

Вывод: 

(array([4, 3]), array([3, 0]))
(array([4, 3]), array([1, 3]))
(array([4, 3]), array([3, 3]))
(array([3, 0]), array([4, 3]))
(array([3, 0]), array([1, 3]))
(array([3, 0]), array([3, 3]))
(array([1, 3]), array([4, 3]))
(array([1, 3]), array([3, 0]))
(array([1, 3]), array([3, 3]))
(array([3, 3]), array([4, 3]))
(array([3, 3]), array([3, 0]))
(array([3, 3]), array([1, 3]))

Итак, в вашем примере вы передали бы 4 в качестве второго аргумента permutations. Я должен предупредить вас, что вы получите колоссальные 15 249 024 перестановок.
РЕДАКТИРОВАТЬ:
Чтобы получить комбинации, а НЕ перестановки, вы должны заменить permutations на combinations.Даже в этом случае вы получите 635 376 результатов.Берегитесь перед печатью!

...