python中遍历list删除元素引发的问题与解决办法
引发问题的场景
今天在写一个小游戏的demo时,进行游戏元素操作时,遇到了一个问题.类似下面代码:
list = ['a','b','c','d'] # element_type == list
for i in list:
print('元素的下标为{},元素的值{}'.format(list.index(i),list)) # 打出内容.方便查看
list.remove(i)
print(list)
本意是遍历删除list中的所有元素.最后list应该为一个空数组.
但是代码实际运行结果是
元素的下标为0,元素的值['a', 'b', 'c', 'd']
元素的下标为1,元素的值['b', 'c', 'd']
['b', 'd'] # 余下两个元素
问题分析
很明显的是,在list中只删除了前两个元素['a','b'],但是本意是删除所有元素.
也就是for循环是执行了两次,而非四次.
- 我猜想问题可能是这样的:
- 在进入for循环的时候,循环下标是递增的.
- 而在for循环操作list的时候(上面代码的删除),list中的元素是动态发生变化的.(当list.remove(list_elem)的时候,len(list)已经发生变化.)
循环下标递增,而循环元素在动态变化.就造成了上面的结果.
换一种循环遍历的方式:
for i in range(len(list)):
list.remove(list[i])
如上面的代码,如果强制指定i,就会报错:下标超出.这是因为强制指定了循环次数.
解决办法
遍历在新的列表操作，删除是在原来的列表操作
a = [1,2,3,4,5,6,7,8]
print(id(a)) # 两个元素的id不相同
print(id(a[:])) # 两个元素的id不相同
for i in a[:]:
if i>5:
pass
else:
a.remove(i)
print(a)
print('-------------------------')
print(id(a))
