Python--遍歷列表時刪除元素的正確做法
阿新 • • 發佈:2018-12-20
一.問題描述
這是在工作中遇到的一段程式碼,原理大概和下面類似(判斷某一個元素是否符合要求,不符合刪除該元素,最後得到符合要求的列表):
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)
執行結果: