python列表过滤

python列表过滤

写代码时会遇到这样的需求：从指定列表中删除不符合要求的元素。这时直观的做法就是遍历整个列表，遇到不符合要求的就删除。但是下面这种写法是错的：

for item in list:
    if not_wanted(item):
        list.remove(item)
    return list

出错的原因是python不允许在遍历一个列表的同时删除里面的元素，这会造成某些元素无法被遍历到。

正确的做法应该是下面这样：

def filter_list(l):
    return [item for item in l if is_wanted(item)]

这段代码的意思是将所有符合要求的元素组成一个新的列表并返回。当然下面的写法也是可以的：

def filter_list(l):
    return filter(is_wanted, l)