1. 程式人生 > >Python--遍歷列表時刪除元素的正確做法

Python--遍歷列表時刪除元素的正確做法

一.問題描述

 這是在工作中遇到的一段程式碼,原理大概和下面類似(判斷某一個元素是否符合要求,不符合刪除該元素,最後得到符合要求的列表):

a = [1,2,3,4,5,6,7,8]
for i in a:
    if i>5:
        pass
    else:
        a.remove(i)
    print(a)

執行結果:

             

二.問題分析

因為刪除元素後,整個列表的元素會往前移動,而i卻是在最初就已經確定了,是不斷增大的,所以並不能得到想要的結果。

三.解決方法

1.遍歷在新的列表操作,刪除是在原來的列表操作

a = [1,2,3,4,5,6,7,8]
print(id(a)) 
print(id(a[:])) 
for i in a[:]:
    if i>5:
        pass
    else:
        a.remove(i)
    print(a)
print('-------------------------')
print(id(a))

執行結果:

2.filter

內建函式filter()官方文件參考:https://docs.python.org/3/library/functions.html#filter

a = [1,2,3,4,5,6,7,8]
b = filter(lambda x: x>5,a)
print(list(b))

執行結果:

3.列表解析

a = [1,2,3,4,5,6,7,8]
b = [i for i in a if i >5]
print(b)

執行結果:

4.倒序刪除

因為列表總是“向前移”,所以可以倒序遍歷,即使後面的元素被修改了,還沒有被遍歷的元素和其座標還是保持不變的。

a = [1,2,3,4,5,6,7,8]
print(id(a))
for i in range(len(a)-1,-1,-1):
    if a[i] > 5:
        pass
    else:
        a.remove(a[i])
print(id(a))
print('--------------------')
print(a)

執行結果: