Python列表边遍历边修改问题解决方案:alist[:]

Posted Pangolin2

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Python列表边遍历边修改问题解决方案:alist[:]相关的知识,希望对你有一定的参考价值。

最近在看python,遇到个简单的问题:删除列表中指定的重复元素,发现一些实用并且有趣的东西。

1.错误示范

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

运行结果:[2, 2, 3, 3, 2, 2, 1, 1]
错误原因:删除列表元素,导致列表内容改变,部分元素位置前移;当继续进行for循环时,索引继续加一,导致跳过一个元素。
本例中,第二个“1”和第四个“1”被跳过,但是remove()是删除从列表第一个出现的元素,所以当第三个“1”出现进入if语句时,第二个“1”被删除。

2.解决方法一:alist[1]

alist = [1,1,2,2,3,3,2,2,1,1]
for i in alist[:]:                ###
    if i ==1:
        alist.remove(1)
print(alist)

运行结果:[2, 2, 3, 3, 2, 2]
简析:将for循环中的控制语句改成" for i in alist[:]"后,程序正确实现功能。对于这个很神奇的alist[:],查了好久,都没有找到相关资料。后来,在和朋友的讨论中,我突然想到“地址”的抽象概念。错误实现的语句" for i in alist"中的"alist",是列表的名称,类似于C中的数组名,其实可以看做一个指向该列表的指针,执行for循环时不断地通过这个地址找到对应元素,导致出现错误结果。
然而," for i in alist[:]"中,“alist[:]”是将alist中的所有元素取出,存储在另一块独立与alist的区域中,这样的话,当修改原列表alist时,并不会修改for循环的判断逻辑,其实就是相当于alist复制给一个新的列表,利用新的列表进行for循环的控制。
为验证以上猜想,用id()命令分别查看alist 和 alist[:]的地址,果然不同。

3.解决方案二:定义一个新的列表

alist = [1,1,2,2,3,3,2,2,1,1]
alist_new = []
for i in alist:
    if i!=1:
        alist_new.append(i)
alist = alist_new
print  (alist)

运行结果:[2, 2, 3, 3, 2, 2]
简析:定义一个新的列表,将不满足条件的值添加到新表中,遍历原列表后,将新列表覆盖原列表。
缺点:新列表会占用内存

4.解决方法三:改变循环条件

alist = [1,1,2,2,3,3,2,2,1,1]
i = 0
while i<len(alist):
    if alist[i]==1:
        alist.pop(i)
        i = i-1
    i = i+1
print (alist)

运行结果:[2, 2, 3, 3, 2, 2]
简析:改变循环条件,判断是否遍历完成,如果删除一个元素,后面元素前移,此时令循环条件i减1,再次判断当前位置,正好判断的是后移过来的新元素。直到所有元素全部遍历。
缺点:逻辑稍复杂,比较优化

5.解决方法四:利用while循环和in、index()函数

alist = [1,1,2,2,3,3,2,2,1,1]
while 1 in alist :
    alist.pop(alist.index(1))
print (alist) 

运行结果:[2, 2, 3, 3, 2, 2]

简析:(1)关键字in,判断一个元素是否在一个集合中,返回True/False;同理,not in也是判断一个元素是否不在一个集合中,返回True/False;有一个类似的关键字 is,判断两个变量是否是同一对象,即同一内存空间,或者地址相同,同样有not is。
(2)index()方法,返回元素在序列中第一次出现的索引号,如果元素不存在于序列中,则会报出异常。
(3)pop() 函数用于移除列表中的一个元素(可用需要删除的索引号作为输入参数,默认删除最后一个元素),并且返回该元素的值。
(4)逻辑:在while条件中不断判断列表中是否有目标元素,若有,则用index()找到该元素的索引号,并用pop()删除,直到删除干净。

6.解决方法五:利用filter()和labmbda

alist = [1,1,2,2,3,3,2,2,1,1]
alist = list(filter(lambda x: x!=1, alist))
print (alist)

运行结果:[2, 2, 3, 3, 2, 2]
简析:(1)根据题目要求,就是对列表进行过滤处理,很自然地想到了filter()函数,配合lambda表达式简直完美。
(2)filter()函数:用于过滤序列,过滤掉不符合条件的元素,返回由符合条件元素组成的新列表。该接收两个参数,第一个为函数,第二个为序列,序列的每个元素作为参数传递给函数进行判,然后返回 True 或 False,最后将返回 True 的元素放到新列表中。
用法:filter(function, iterable)
(3)lambda:匿名定义简单函数,主体是一个表达式,常用于函数式编程。
用法:lambda [arg1 [,arg2,.....argn]]:expression
(4)逻辑:用lambda定义匿名函数,判断元素是否为非“1”,若真则返回True,否则返回False。在用filter函数进行过滤处理,实现对列表的过滤功能。

以上是关于Python列表边遍历边修改问题解决方案:alist[:]的主要内容,如果未能解决你的问题,请参考以下文章

Python面试题目之(针对dict或者set数据类型)边遍历 边修改 报错dictionary changed size during iteration

Java中ArrayList边遍历边修改

List实现边查边增删改

python循环修改列表注意事项

NX二次开发(Python)-建模-遍历模型中的体面边对象

找到 MST 的关键边:可以使用修改后的 Prim 算法吗?