例如,我有 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
最佳答案
对于像你拥有的这样大的东西,打印结果非常困难,所以我会给你一个较小的东西的工作示例 -
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 个结果。打印前请注意!
关于python - 矩阵行组合的生成,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52727099/