Question

例如，我有64行矩阵。我想让所有组合看起来像我们可以从行0到3取1个元素，从4到7取1个元素，依此类推。所以我需要矩阵中的16行。我的想法是制作4行16矩阵，并尝试从每行中选择一个。但是我的代码不起作用。任务是获取所有行的组合，其中我们从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的排列。
编辑：
要获得组合而不是排列，您可以将permutations替换为combinations。即使在这种情况下，您也可以获得635,376个结果。打印前要小心！

矩阵行组合的生成

1 个答案: