Python使用.remove()删除子列表元素未按预期生效问题

问题本质 #

在正向遍历列表的过程中直接修改原列表（执行删除操作），会触发列表元素的索引位移，导致遍历过程跳过部分元素，最终无法删除所有符合条件的0。

以第一个子列表[0, 0, 0, 0, 0, 1]为例，执行过程如下：

• 初始状态：子列表索引0~5依次对应值0,0,0,0,0,1，遍历指针从索引0启动
• 指针指向索引0：检测到值为0，调用remove(0)删除第一个匹配的0，删除后列表变为[0,0,0,0,1]，后续所有元素整体向前移动1位；此时遍历指针自动向后移动到索引1，原本在索引1位置的0因为位移到了索引0，直接被跳过
• 指针指向索引1：当前位置值为0（原索引2位置的元素），再次执行删除，列表变为[0,0,1]，后续元素再次前移，指针继续后移到索引2
• 指针指向索引2：当前值为1，不满足删除条件，该子列表遍历结束
• 最终剩余2个0未被删除，和你运行得到的结果一致。

正确实现方式 #

方法1：倒序遍历删除（操作原列表） #

倒序遍历的时候，删除元素只会影响已经遍历过的位置的索引，不会干扰未遍历部分的元素位置，不会出现漏删：

lst = [[0, 0, 0, 0, 0, 1], [1, 2, 3, 4, 5, 4, 3, 0], [1, 4, 5, 6, 7, 0]]

for week in lst:
    for i in range(len(week)-1, -1, -1):
        if week[i] == 0:
            week.pop(i)

print(lst)
# 输出：[[1], [1, 2, 3, 4, 5, 4, 3], [1, 4, 5, 6, 7]]

方法2：列表推导式生成新列表（推荐写法） #

不需要操作原列表，直接过滤掉值为0的元素生成新列表，逻辑清晰无索引错位问题：

lst = [[0, 0, 0, 0, 0, 1], [1, 2, 3, 4, 5, 4, 3, 0], [1, 4, 5, 6, 7, 0]]
lst = [[day for day in week if day != 0] for week in lst]

print(lst)
# 输出：[[1], [1, 2, 3, 4, 5, 4, 3], [1, 4, 5, 6, 7]]

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