python - 矩阵行组合的生成

Question

例如，我有 64 行的矩阵。我想要得到所有的组合，看起来我们可以从第 0 行到第 3 行中取出 1 个元素，从第 4 行到第 7 行中取出 1 个元素，等等。所以我需要矩阵中的 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

Answer 1

对于像你拥有的这样大的东西，打印结果非常困难，所以我会给你一个较小的东西的工作示例 -

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 种排列。
编辑:
要获取组合而不是排列，您可以将 排列 替换为 组合。即使在这种情况下，您也会得到 635,376 个结果。打印前请注意!