Python中for循环遍历列表时增删元素不符合预期的问题求助

为什么现有代码不生效？ #

1. 移除1不彻底：遍历列表时直接调用remove()修改原列表，会导致迭代器跳过元素。比如原列表[1,1,5]，第一次循环移除第一个1后，列表变为[1,5]，迭代器的下一个位置指向原列表的第二个元素（现在是新列表的第一个元素1），但迭代器已经走到索引1，直接取到5，跳过了剩下的1。
2. 添加6无效：原列表里根本没有6，第二个if条件永远不会触发，自然不会执行append()。

修正后的代码示例 #

方法1：使用列表推导式（推荐，简洁高效） #

import random

def calculate_score():
    worplist = [1, 1, 5]
    # 生成不包含1的新列表
    worplist = [num for num in worplist if num != 1]
    # 处理6的逻辑：遍历列表副本，避免修改原列表影响迭代
    for num in worplist.copy():
        if num == 6:
            worplist.append(num)
    
    print(sum(worplist))
calculate_score()

方法2：倒序遍历原列表（适合需要直接修改原列表的场景） #

import random

def calculate_score():
    worplist = [1, 1, 5]
    # 从后往前遍历，移除元素时不会打乱前面未遍历的元素索引
    for i in range(len(worplist)-1, -1, -1):
        if worplist[i] == 1:
            worplist.pop(i)
        # 若存在6，执行添加操作
        if worplist[i] == 6:
            worplist.append(worplist[i])
    
    print(sum(worplist))
calculate_score()

测试说明 #

如果要验证添加6的逻辑，可以把初始worplist改成[1,6,5]，两种方法都会正确添加额外的6到列表末尾。

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