Python语法糖——遍历列表时删除元素

Python的for能够遍历一个List，可是在遍历的过程当中删除元素经常会获得意想不到的结果甚至程序出现异常，例如：

lst = [1, 1, 0, 2, 0, 0, 8, 3, 0, 2, 5, 0, 2, 6]

for item in lst:
    if item == 0:
        lst.remove(item)
print lst

输出： [1, 1, 2, 8, 3, 2, 5, 0, 2, 6]

这段代码的本意是想删除列表中元素为 0 的项，但实际跑起来并非那个结果。

再看下面这个例子，利用索引来遍历删除列表中的元素：

for item in range(len(lst)):
    if lst[item] == 0:
        del lst[item]
print lst

输出： IndexError: list index out of range

这时候就报错了，抛出了数组下标越界的异常。缘由是用for发起任何形式的遍历时，它的遍历顺序都是从最初就肯定的，而在遍历中删除了元素会致使当前索引的变化，这样一是会致使漏删元素，二是会致使遍历超过链表的长度。

因此遍历一个List时要删除其中的部分元素就须要其余的解决方案。

方法一：能够使用filter过滤返回新的List

lst = [1, 1, 0, 2, 0, 0, 8, 3, 0, 2, 5, 0, 2, 6]

lst = filter(lambda x: x != 0, lst)
print lst

这样能够安全删除列表中值为 0 的元素了，filter包括两个参数，分别是function和list，filter把传入的函数依次做用于每一个元素，而后根据返回值是True仍是False来决定是保留仍是丢弃该元素。

方法二：列表解析

lst = [x for x in lst if x != 0]
print lst

方法三：或者遍历拷贝的List，操做原始的List

for item in lst[:]:
    if item == 0:
        lst.remove(item)
print lst

方法四：用while循环来搞定，每次循环都先会判断 0 是否在列表中

while 0 in lst:
    lst.remove(0)
print lst

方法五：倒序循环遍历

for item in range(len(lst) - 1, -1, -1):
    if lst[item] == 0:
        del lst[item]
print lst

倒序是最geek的解决办法，可是倒序的代码可读性差，推荐用filter，这样写出来的代码更Pythonic。