Python：使用for循环移动列表所有零至末尾遇问题求助

我想实现将列表中所有零移至末尾的功能，采用for循环与Python内置函数编写了如下代码：

def move_zeros(lst):
    new_ls = []
    for i in lst:
        if i == 0:
            new_ls.append(i)
            lst.remove(i)
    return (lst)

samp_lis = [9, 0, 0, 9, 1, 2, 0, 1, 0, 1, 0, 3, 0, 1, 9, 0, 0, 0, 0, 9]
print(move_zeros(samp_lis))

运行后发现for循环仅执行到索引13就结束了，无法处理完所有元素。
输入列表：
[9, 0, 0, 9, 1, 2, 0, 1, 0, 1, 0, 3, 0, 1, 9, **0**, 0, 0, 0, 9]

得到的结果：

[9, 9, 1, 2, 1, 1, 3, 1, 9, 0, 0, 0, 9, 0, 0, 0, 0, 0, 0, 0]

之前看到类似问题多采用排序方案解决，我想确认是否可以用for循环实现该需求。

问题原因分析 #

你的代码存在两个核心问题：

1. 遍历过程中修改原列表导致迭代异常：Python的for循环依赖列表的迭代器，当你在遍历期间调用lst.remove(i)删除元素时，列表长度缩短，后续元素会向前移位，而迭代器的索引不会回退，直接跳过了原本位于被删除元素后方的元素，最终导致循环提前终止，无法处理所有零。
2. 未将收集的零合并回结果：你把筛选出的零存在了new_ls中，但返回时只返回了修改后的lst，没有将new_ls里的零追加到末尾，这也导致结果不符合预期。

用for循环实现的正确方案 #

方案一：遍历原列表的副本，避免修改影响遍历

通过遍历原列表的副本，确保迭代过程不受原列表修改的影响，统计零的数量后统一追加到末尾：

def move_zeros(lst):
    zero_count = 0
    # 遍历原列表的副本，避免修改原列表干扰遍历流程
    for i in lst.copy():
        if i == 0:
            lst.remove(i)
            zero_count += 1
    # 将所有零追加到列表末尾
    lst.extend([0] * zero_count)
    return lst

samp_lis = [9, 0, 0, 9, 1, 2, 0, 1, 0, 1, 0, 3, 0, 1, 9, 0, 0, 0, 0, 9]
print(move_zeros(samp_lis))
# 输出：[9, 9, 1, 2, 1, 1, 3, 1, 9, 9, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]

方案二：分类收集元素后合并（更高效）

直接遍历原列表，将非零元素和零分别存入两个新列表，最后合并返回，这种方式避免了列表删除操作（中间删除元素的时间复杂度为O(n)），效率更高：

def move_zeros(lst):
    non_zeros = []
    zeros = []
    for i in lst:
        if i == 0:
            zeros.append(i)
        else:
            non_zeros.append(i)
    return non_zeros + zeros

samp_lis = [9, 0, 0, 9, 1, 2, 0, 1, 0, 1, 0, 3, 0, 1, 9, 0, 0, 0, 0, 9]
print(move_zeros(samp_lis))
# 输出：[9, 9, 1, 2, 1, 1, 3, 1, 9, 9, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]

内容的提问来源于Stack Exchange，提问作者Christian