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]