在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]