Python遍歷列表時刪除元素

2021-09-12 00:06:49 字數 1797 閱讀 3512

本文首發於我的個人部落格suixin』s blog
python的for迴圈可以遍歷乙個列表,但若需要同時將符合條件的元素刪除時,則會出錯。

s =[1

,2,3

,4,1

,1]s1 = s

for i in s1:

if i ==1:

s1.remove(i)

print

(s1)

輸出:

[2, 3, 4, 1]
另一種:

s2 = s

for idx in

range

(len

(s2)):

if s2[idx]==1

:del s2[idx]

print

(s2)

輸出:

traceback (most recent call last):

file "temp.py", line 11, in if s2[idx] == 1:

indexerror: list index out of range

python中用for發起任何形式的遍歷時,它的遍歷順序都是從最初就確定的,而在遍歷中刪除了元素會導致當前索引的變化,所以遍歷中間取到的索引的那個值已經變了,會漏掉一些元素。另一種錯誤是刪除元素後列表長度縮小了,所以會超出索引範圍。

使用filter()函式過濾掉符合指定條件的元素。

s3 = s

print

(list

(filter

(lambda x: x !=

1, s3)

))

此處lambda為python中函式的簡化定義形式。

使用列表解析式。

s4 =

[i for i in s if i !=1]

print

(s4)

把原始列表拷貝給乙個新列表,遍歷新列表,修改原始列表(或相反)。

s6 = s

for i in s6[:]

:if i ==1:

s6.remove(i)

print

(s6)

但是,下面的賦值操作給新列表是不行的,因為新變數和原變數的實體地址是相同的,可通過id()函式檢視。

s5 = s

for i in s:

if i ==1:

s5.remove(i)

print

(s5)

可通過深拷貝解決上述問題:

import copy

s5 = copy.deepcopy(s)

for i in s:

if i ==1:

s5.remove(i)

print

(s5)

while迴圈來做判斷條件。

s7 = s

while

1in s7:

s7.remove(1)

print

(s7)

上述4種方法的輸出皆為:[2, 3, 4]

Python語法糖 遍歷列表時刪除元素

python的for可以遍歷乙個list,但是在遍歷的過程中刪除元素常常會得到意想不到的結果甚至程式出現異常,例如 lst 1,1,0,2,0,0,8,3,0,2,5,0,2,6 for item in lst if item 0 lst.remove item print lst 輸出 1,1,2...

Python遍歷列表刪除多個列表元素

在遍歷list的時候,刪除符合條件的資料,結果不符合預期 num list 1,2,2,2,3 print num list for item in num list if item 2 num list.remove item else print item print num list 結果是 ...

Python遍歷列表刪除多個列表元素

在遍歷list的時候,刪除符合條件的資料,結果不符合預期 num list 1,2,2,2,3 print num list for item in num list if item 2 num list.remove item else print item print num list 結果是 ...