突然记起来一年前刚入行时,领导面试我时提的问题,删除list中所有等于某个值的元素
比如 删除 [1,2,3,4,1,2,3,4,1,3,4] 中所有的2元素
删除元素常用 remove, pop 方法,
remove可以直接删除某个值:
list.remove(1)
但是每次只能删除第一个,且如果list中不存在这个值则会抛出异常。
pop可以根据索引来删除:
list.pop(2)
但是同样在此处不适用,因为 pop 都是直接作用于list对象,
当我们在遍历list中,直接删除会对遍历有影响。
当时我给出的方案是建立一个新list,遍历原list,如果不等于这个值,则放入新list中。。。
所以落下一个 虽然不会,但是敢想 的评价。。
当时事后回想,觉得应该是有两种方案:
1.死循环无限remove,遇到异常则return
2.从后向前遍历pop,遇到异常return
但是现在想起来,完全可以有更好的方法:
filter函数
def remove_item(n):
return n != 1
aaa = [1,2,3,5,6,8,9,5,3,1]
print list(filter(remove_item, aaa))
同样,用于去除list中的空字符串或者None同样好用:
new_list = list(filter(None, list))