python 使用for循环删除列表元素,容易出现结果与期望不一致问题。
例如代码:要删除列表中的3或者4,但是最后结果是4未删除掉。
test_list = [1, 2, 3, 4, 5]
print(test_list)
for ele in test_list:
if ele == 3 or ele == 4:
test_list.remove(ele)
else:
print(ele)
print(test_list)
结果:
[1, 2, 3, 4, 5]
1
2
5
[1, 2, 4, 5]
因为删除3后,3之后的[4,5]前移,4占了3原来的位置。下次循环是取3之后的数,即现在4之后的数(因为4去3的位置顶替了),也就是继续去判断5了。
test_list=[1,2, 3, 4, 5]
i=0
#不能使用for i in range(0,len(num_list))会导致最后index溢出
while i < len(test_list):
if test_list[i] == 3 or test_list[i]==4:
test_list.pop(i)
i-=1
else:
print("num_list[{}]={}".format(i,test_list[i]))
i+=1
print(test_list)
结果:
num_list[0]=1
num_list[1]=2
num_list[2]=5
[1, 2, 5]
注意:while循环不可以换成for,for i in range(0,len(num_list))会导致最后索引i溢出
切片是生成数组的copy方式之一
test_list = [1, 2, 3, 4, 5]
# num_list[:]是对num_list的拷贝
for item in test_list[:]:
if item == 3 or item==4:
test_list.remove(item)
else:
print(item)
print(test_list)
结果:
1
2
5
[1, 2, 5]
倒叙遍历,删除某个元素后,后面元素向前顶替位置,后面的元素是已经check过的,所以不会导致元素遗漏问题。
test_list = [1, 2, 3, 4, 5]
for i in range(len(test_list)-1, -1, -1):
if test_list[i] == 3 or test_list[i] == 4:
test_list.pop(i)
else:
print("num_list[{}]={}".format(i,test_list[i]))
print(test_list)
结果:
num_list[4]=5
num_list[1]=2
num_list[0]=1
[1, 2, 5]
test_list.pop(i)
test_list.remove(item)
test_list2=[1,2,3,4,5,6,7,5]
test_list2.remove(5)
print(test_list2)
结果:
[1, 2, 3, 4, 6, 7, 5]
test_list2=[1,2,3,4,5,6,7,5]
del test_list2[1]
print(test_list2) #[1, 3, 4, 5, 6, 7, 5]
test_list2=[1,2,3,4,5,6,7,5]
del test_list2[2:4] #删除3,4
print(test_list2) #[1, 2, 5, 6, 7, 5]
test_list2=[1,2,3,4,5,6,7,5]
del test_list2
print(test_list2) #NameError: name 'test_list2' is not defined