如何实现Python列表中各元素的独立并行循环移位迭代

需求说明 #

每次迭代时，列表内的每个字符串元素需同时进行原地循环左移（示例为左移，可按需调整为右移），例如：
初始列表：

circle_elem = ['abcde', 'fghij', 'klmno', 'pqrst', 'vwxyu']

迭代后依次得到：

['bcdea', 'ghijf', 'lmnok', 'qrstp', 'wxyuv']
['cdeab', 'hijfg', 'mnokl', 'rstpq', 'xyuvw']
# ... 循环往复

已知所有字符串元素长度一致。

问题分析 #

你之前尝试的方法均未针对单个字符串内部的字符移位操作：

• circular_shifts 是对整个列表的元素顺序进行循环，而非元素内部字符；
• itertools.cycle(circle_elem) 是循环迭代列表中的每个元素，不涉及元素内部移位；
• itertools.product(circle_elem) 生成的是列表元素的笛卡尔积，与移位需求无关。

解决方案 #

方案1：无限迭代的生成器（持续输出移位结果）

通过为每个字符串生成独立的循环移位序列，每次迭代同步取出每个序列的下一个元素，实现无限循环的并行移位：

from itertools import cycle

def cycle_elements(lst):
    elem_length = len(lst[0])
    # 为每个字符串生成无限循环的移位序列
    shifted_gens = [cycle(s[1:] + s) for s in lst]
    while True:
        yield [next(gen) for gen in shifted_gens]

# 使用示例
circle_elem = ['abcde', 'fghij', 'klmno', 'pqrst', 'vwxyu']
shift_iter = cycle_elements(circle_elem)

# 输出前3次移位结果
for _ in range(3):
    print(next(shift_iter))

输出结果：

['bcdea', 'ghijf', 'lmnok', 'qrstp', 'wxyuv']
['cdeab', 'hijfg', 'mnokl', 'rstpq', 'xyuvw']
['deabc', 'ijfgh', 'noklm', 'stpqr', 'yuvwx']

方案2：生成所有可能的移位状态（有限迭代）

由于每个字符串长度为n，最多有n种不同的移位状态，可一次性生成所有结果：

def all_cycle_shifts(lst):
    elem_length = len(lst[0])
    return [
        [s[shift:] + s[:shift] for s in lst]
        for shift in range(elem_length)
    ]

# 使用示例
circle_elem = ['abcde', 'fghij', 'klmno', 'pqrst', 'vwxyu']
for shifted_list in all_cycle_shifts(circle_elem):
    print(shifted_list)

输出会包含从原列表到左移4位的所有状态（共5种）。

调整移位方向（右移）

若需要右移一位，只需将移位逻辑改为 s[-1] + s[:-1]，例如方案1中修改生成器：

shifted_gens = [cycle(s[-1] + s[:-1]) for s in lst]

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