Python列表按条件删除元素：循环仅移除首个符合项的问题排查

问题描述 #

需求是移除列表中包含'strategies available for player'或等于'\n'的元素，但用普通for循环遍历并删除后，只能移除第一个符合条件的'strategies available for player'实例，第二个完全被跳过。

原代码如下：

a_list = ['-2 strategies available for player-\n',
 'd1 = c(0.0216,0.0519,0.0714,0.0942,0.1050);\n',
 'd2 = c(0.0382,0.0475,0.0526,0.0768,0.1173);\n',
 'd3 = c(0.0297,0.0561,0.0822,0.0834,0.1321);\n',
 'd4 = c(0.1179,0.1233,0.1351,0.1369,0.1669);\n',
 'd5 = c(0.0143,0.0256,0.0294,0.0366,0.0461);\n',
 'd6 = c(0.03300,0.0535,0.0832,0.0867,0.1014);\n',
 'd7 = c(0.0661,0.0921,0.1205,0.1398,0.1650);\n',
 'd8 = c(0.0629,0.08316,0.1210,0.1467,0.1642);\n',
 '\n',
 '-3 strategies available for player-\n',
 'd1 = c(0.0594,0.0691,0.0797,0.1020,0.1134);\n',
 'd2 = c(0.0613,0.0737,0.1075,0.1160,0.1299);\n',
 'd3 = c(0.1082,0.1216,0.1343,0.1410,0.1495);\n',
 'd4 = c(0.0949,0.1086,0.1288,0.1506,0.1583);\n',
 'd5 = c(0.0371,0.0498,0.0571,0.0688,0.0961);\n',
 'd6 = c(0.0752,0.0962,0.1056,0.1218,0.1465);\n',
 'd7 = c(0.0849,0.1209,0.1321,0.1574,0.1663);\n',
 'd8 = c(0.0737,0.1216,0.1498,0.1793,0.1923);']

for el in a_list:
    if 'strategies available for' in el or el == '\n':
        a_list.remove(el)

for el in a_list:
    print(el)

原因分析 #

这是因为在遍历列表的同时直接修改列表（比如调用remove()）会打乱迭代器的遍历顺序。

具体来说，for循环是基于列表的初始迭代器逐个获取元素的：

1. 当你移除第一个目标元素后，列表长度减1，后续所有元素都会往前移动一个位置；
2. 但循环的迭代器不会回溯，会继续取下一个“下标对应的元素”，导致原本在被移除元素后方的元素被跳过，也就错过了第二个'strategies available for player'元素。

比如原列表中'\n'初始在索引9，移除第一个目标元素（索引0）后，'\n'的索引变为8，当循环走到索引9时，已经跳过了这个位置；而第二个目标元素原本在索引10，现在变为9，循环到这里时已经处理过索引9，所以不会被检查到。

解决方法 #

推荐三种可靠的解决方式：

方法1：遍历列表的副本 #

遍历原列表的切片副本（a_list[:]），这样修改原列表不会影响遍历过程：

for el in a_list[:]:  # 遍历列表副本
    if 'strategies available for' in el or el == '\n':
        a_list.remove(el)

for el in a_list:
    print(el)

方法2：使用列表推导式（最简洁） #

直接生成新列表，只保留符合要求的元素，这是Python处理此类需求的常用方式：

a_list = [el for el in a_list if not ('strategies available for' in el or el == '\n')]

for el in a_list:
    print(el)

方法3：反向遍历列表 #

从列表末尾往前遍历，移除元素不会影响前方未遍历元素的位置：

for i in range(len(a_list)-1, -1, -1):
    if 'strategies available for' in a_list[i] or a_list[i] == '\n':
        del a_list[i]

for el in a_list:
    print(el)

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