Python实现列表元素特定排列：位置元素不重复且适配任意长度

我有一个列表A，需要生成特定排列，要求所有排列结果中，没有任何位置被同一元素重复占据。目前使用的代码、运行输出以及期望输出如下，同时要求代码能够适配任意长度的A。

当前使用代码 #

from itertools import permutations

A = [1, 2, 3]

# 生成所有可能的排列并将元组转为列表
all_permutations = [list(perm) for perm in permutations(A)]

# 打印排列
for perm in all_permutations:
    print(perm)

当前输出 #

[1, 2, 3]
[1, 3, 2]
[2, 1, 3]
[2, 3, 1]
[3, 1, 2]
[3, 2, 1]

期望输出 #

[1,2,3]
[3,1,2]
[2,3,1]

解决方案 #

你需要的是循环移位排列，这类排列能保证每个位置上的元素在所有结果中不重复，且适配任意长度的列表。以下是实现代码：

def generate_cyclic_permutations(lst):
    n = len(lst)
    # 生成所有循环移位后的排列
    return [lst[i:] + lst[:i] for i in range(n)]

# 示例调用
A = [1, 2, 3]
result = generate_cyclic_permutations(A)
for perm in result:
    print(perm)

运行这段代码会输出你期望的结果，对于任意长度的输入列表，它都会生成对应数量的合法排列，完全满足每个位置无重复元素的要求。

内容的提问来源于stack exchange，提问作者Dbatt