Python列表循环移动元素函数全局修改问题求助

你想要实现一个循环移动列表元素的函数：移除列表第一个元素并放到末尾，但当前代码会直接修改原列表，导致断言失败。先看你代码的问题：

原代码问题分析 #

第一个版本直接对传入的列表执行pop(0)和append，这两个方法都是直接修改原列表的操作。第一次调用cycle(l)后，原列表l已经变成[2,3,1]，嵌套调用cycle(cycle(l))时，处理的是修改后的列表，自然不符合预期，最后断言原列表等于初始值也会失败。

第二个版本的错误更明显：

• z=y.copy是引用了列表的copy方法对象，不是调用方法，正确写法是z=y.copy()
• return语句之后的代码y=z永远不会执行，函数在return时就已经结束，根本没法恢复原列表

正确解决方案 #

方案1：操作列表副本 #

在函数内部创建原列表的副本，所有操作都在副本上进行，原列表不会被修改：

l = [1,2,3]
def cycle(y):
    # 创建原列表的副本，避免修改原列表
    new_list = y.copy()  # 也可以用 new_list = y[:]，效果相同
    first_element = new_list.pop(0)
    new_list.append(first_element)
    return new_list

assert(cycle(l) == [2, 3, 1])
assert(cycle(cycle(l)) == [3, 1, 2])
assert(l == [1, 2, 3])  # 原列表保持初始状态，断言通过

方案2：切片拼接（更简洁） #

利用列表切片直接生成新列表，完全不需要修改原列表：

l = [1,2,3]
def cycle(y):
    # 空列表直接返回，避免索引错误
    if not y:
        return y
    # 从第二个元素开始的切片 + 第一个元素的列表，拼接成新列表
    return y[1:] + [y[0]]

assert(cycle(l) == [2, 3, 1])
assert(cycle(cycle(l)) == [3, 1, 2])
assert(l == [1, 2, 3])

这种写法更简洁，而且切片操作本身返回的就是新列表，不会对原列表产生任何修改。

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