本篇文章小编给大家分享一下Python遍历列表时删除元素代码示例,文章代码介绍的很详细,小编觉得挺不错的,现在分享给大家供大家参考,有需要的小伙伴们可以来看看。
tk在科学养猪群里问bluerust、scz是否碰上过这个Python坑,
示例1:
bas = [ 'ba1', 'ba2', 'ba3', 'ba4', 'ba5' ] for ba in bas : print( ba ) if ( ba.find( 'ba' ) != -1 ) : bas.remove( ba ) print( bas ) print( bas )
即遍历list的过程中动态删除元素。
上述代码输出如下:
ba1 ['ba2', 'ba3', 'ba4', 'ba5'] ba3 // ba2被跳过去了 ['ba2', 'ba4', 'ba5'] ba5 // ba4被跳过去了 ['ba2', 'ba4'] ['ba2', 'ba4'] // 列表未删干净
对示例1做点改动,
示例2:
bas = [ 'ba1', 'ba2', 'ba3', 'ba4', 'ba5' ] for i in range( len( bas ) ) : print( i ) print( bas[i] ) if ( bas[i].find( 'ba' ) != -1 ) : del bas[i] print( bas ) print( bas )
上述代码输出如下:
0 ba1 ['ba2', 'ba3', 'ba4', 'ba5'] 1 ba3 ['ba2', 'ba4', 'ba5'] 2 ba5 ['ba2', 'ba4'] 3 Traceback (most recent call last): File "", line 3, in IndexError: list index out of range ['ba2', 'ba4']
循环变量i只递增到3,进而抛出IndexError。
该坑的起因是,for循环中i的取值从最开始就固定了,实际上要求list在for循环中保持不变;遍历list的过程中动态删除元素,导致list发生变化,而i仍固执地按原计划递增遍历list,于是漏删元素、索引越界。
对此,wzhvictor给了5种解决方案。
方式1,利用filter函数
bas = [ 'ba1', 'ba2', 'ba3', 'ba4', 'ba5', 'tk' ] bas = list( filter( lambda ba:ba.find( 'ba' ) == -1, bas ) ) print( bas )
方法2,重新构造list
bas = [ 'ba1', 'ba2', 'ba3', 'ba4', 'ba5', 'tk' ] bas = [ba for ba in bas if ba.find( 'ba' ) == -1] print( bas )
方法3,遍历list的拷贝,对原始list进行删除操作
bas = [ 'ba1', 'ba2', 'ba3', 'ba4', 'ba5', 'tk' ] for ba in bas[:] : if ( ba.find( 'ba' ) != -1 ) : bas.remove( ba ) print( bas )
方法4
bas = [ 'ba1', 'ba2', 'ba3', 'ba4', 'ba5', 'tk' ] for ba in bas[:] : if ( ba.find( 'ba' ) != -1 ) : bas.remove( ba ) print( bas )
方法5,倒序遍历
bas = [ 'ba1', 'ba2', 'ba3', 'ba4', 'ba5', 'tk' ] for i in range( len( bas )-1, -1, -1 ) : if ( bas[i].find( 'ba' ) != -1 ) : del bas[i] print( bas )
就tk的示例1而言,方法4其实不适用,方法4适合从list中删除所有特定值。