当前位置:首页 >> 脚本专栏

python 的列表遍历删除实现代码

python的列表list可以用for循环进行遍历,实际开发中发现一个问题,就是遍历的时候删除会出错,例如

l = [1,2,3,4]
for i in l:
 if i != 4:
 l.remove(i)
print l

这几句话本来意图是想清空列表l,只留元素4,但是实际跑起来并不是那个结果。再看下面,利用index来遍历删除列表l

l = [1, 2, 3, 4]
for i in range(len(l)):
 if l[i] == 4:
 del l[i]

print l

这样没问题,可以遍历删除,但是列表l如果变为 l = [1,2,3,4,5]
如果还是按照上面的方法,设想一下,range开始的范围是0-4,中间遍历的时候删除了一个元素4,这个时候列表变成了= [1,2,3,5],这时候就会报错了,提示下标超出了数组的表示,原因就是上面说的遍历的时候删除了元素

所以python的list在遍历的时候删除元素一定要小心
可以使用filter过滤返回新的list

l = [1,2,3,4]
l = filter(lambda x:x !=4,l)
print l

这样可以安全删除l中值为4的元素了,filter要求两个参数,第一个是规则函数,第二个参数要求输入序列,而lambda这个函数的作用就是产生一个函数,是一种紧凑小函数的写法,一般简单的函数可以这么些

或者可以这样

l = [1,2,3,4]
l = [ i for i in l if i !=4]//同样产生一个新序列,复值给l
print l

或者干脆建立新的list存放要删除的元素

l = [1,2,3,4]
dellist = []
for i in l:
 if i == 4:
 dellist.append(i)
for i in dellist:
 l.remove(i)

这样也能安全删除元素

所以要遍历的时候删除元素一定要小心,特别是有些操作并不报错,但却没有达到预期的效果
上面说到产生新序列,赋值等等,用python的id()这个内置函数来看对象的id,可以理解为内存中的地址,所以有个简要说明

如果

l = [1,2,3,4]
ll = l
l.remove(1)
print l//肯定是[2,3,4]
print ll//这里会是什么?

如果用id函数查看的话就发现

print id(l),id(ll)

打印出相同的号码,说明他们其实是一个值,也就是说上面的print ll将和l打印的一样,所以python有这种性质,用的时候注意一下就行了

一般python书中会讲到这个问题
遍历的时候如果想更新列表本身
推荐的做法是使用切片

l = [1, 2, 3, 4]
for i in l[:]
some code

python中循环删除列表中元素时的坑!

循环删除列表中元素时千万别用正序遍历,一定要用反序遍历!

废话不多说,先上案例代码:

def test(data):
  for i in data:
    data.remove(i)
  return data
data = [1, 2, 3]
print(test(data))

面对以上代码,乍一看以为会打印出空列表,因为test函数内通过for的方法将data中的元素都删除了,其实不然,实际输出如下:

[2]

为什么会产生这种结果呢? 我们来深度剖析一下:

原列表在内存中为:

python 的列表遍历删除实现代码

第一次执行到data.remove(i)时将第一个元素‘1'删除,列表变为:

python 的列表遍历删除实现代码

第二次执行到data.remove(i)时i为第二个元素,即‘3',此时将‘3'删除,列表变为:

python 的列表遍历删除实现代码

此时列表已经没有第三个元素了,即退出循环,将[2]返回。

如何解决这个问题呢? 我们可以用倒序删除的方法!

直接上代码:

def test(data):
  for i in data[::-1]:
    data.remove(i)
  return data


data = [1, 2, 3]
print(test(data))

此时再运行,发现输出结果为空列表

[]

倒序删除原理如下:

原列表在内存中为:

python 的列表遍历删除实现代码

第一次执行到data.remove(i)时是将倒数第一个元素‘3'删除,列表变为:

python 的列表遍历删除实现代码

第二次执行到data.remove(i)时i为倒数第二个元素,即‘2',此时将‘2'删除,列表变为:

python 的列表遍历删除实现代码

第三次执行到data.remove(i)时i为倒数第三个元素,即‘1',此时将‘1'删除,列表变为空列表!

总结:正序删除列表中元素时,被删元素后面的值会向前顶,然后导致漏删。倒序删除元素时,被删元素前面的值不会向后靠,所以可以完整的遍历到列表中所有的元素。