python中使用list的时候,通常需要用到移除其中某些元素,而remove函数就正好可以移除元素,所以就会想到循环遍历list,利用remove函数移除元素,例如下面一段代码:
def remove_item(l,n):
for item in l:
if item==n:
l.remove(item)
if __name__ == "__main__":
l = [1,2,3,3,3]
print("before remove l:",l)
remove_item(l,3)
print("after remove l:",l)
上面的一段代码期望实现的是删除list中等于3的元素,运行结果为:
删除之后仍然还有元素3,并没有完全删除,所以这种方法是有问题的,那么问题在哪呢?
那是因为list的遍历是基于下标的,当你删除其中的一个元素的时候,列表实际上已经发生了变化,该元素后面的所有元素都往前移动了一个位置,所以下次遍历的时候就会跳过该元素后面的一个元素,例子中的第二个3就被跳过了,这个3去填了移除的那个3的位置,所以第二个3不会被移除。
解决办法:方法很多,比如可以把不删除的元素重新添加到一个新的list中,也可以先拷贝一份列表备份,然后遍历备份列表,删除的时候就删除原列表,再或者基于索引遍历,当需要删除元素的时候,索引值对应减1,这些方法都可以根据自己需求选择。