Python中使用for循环删除列表元素的正确方法

直接删除往往结果与期望不一致

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了。

python for循环删除元素的正确方法: 

方法1:while循环+i减

使用while循环时刻监测数组长度,控制数组索引i,当i 遇到删除元素时候就往回移动一个元素位置,即减1

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溢出

方法2:使用copy 

切片是生成数组的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]

方法3:倒叙遍历

倒叙遍历,删除某个元素后,后面元素向前顶替位置,后面的元素是已经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]

python删除list元素:

pop方法

是删除第一个或者是指定删除元素的下标。返回被删除的值。

 test_list.pop(i)


remove方法

直接指定要删除的元素值,删除首个匹配的值。

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] 

del 元素方法

可以删除指定下标的元素,删除指定下标范围内的多个元素,删除整个对象

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

物联沃分享整理
物联沃-IOTWORD物联网 » Python中使用for循环删除列表元素的正确方法

发表评论