python - 在 Python 中，如何生成每列和每行只有一个元素的数组的排列？

Question

例如:

假设您有以下数组:

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

并且您想生成这个数组:

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

Answer 1

import itertools

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

for P in itertools.permutations(range(len(A))):
    print [A[p][i] for i,p in enumerate(P)]

打印:

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