矩阵索引组合，但每行一个

Question

我正在从n * m矩阵中提取数据以累积唯一行组合的值

想象

[ [1,2,3],[4,5,6],[7,8,9]]

我会追随要产生的索引

1 + 4 + 7

1 + 4 + 8

1 + 4 + 9

2 + 4 + 7

2 + 4 + 9

...

我已经查看了itertools和numpy，但无法实现拉出组合的方法

Answer 1

Itertools拥有它：

import itertools
x = [ [1,2,3],[4,5,6],[7,8,9]]
for y in list(itertools.product(*x)):
   print y

它给您：

(1, 4, 7)
(1, 4, 8)
(1, 4, 9)
(1, 5, 7)
(1, 5, 8)
(1, 5, 9)
(1, 6, 7)
...
(3, 6, 9)