python 安全地刪除列表元素的方法
前言:
看似簡單的任務,往往隱藏陷阱!
一個常見的任務是在一個列表上迭代,并根據(jù)條件刪除一些元素。本文將展示如何完成該任務的不同方法,同時展示一些需要避免的陷阱。
假設我們需要修改列表a,并且必須刪除所有不是偶數(shù)的項。
首先實現(xiàn)輔助函數(shù)even(x)來確定一個數(shù)字x是否是偶數(shù):
a = [1, 2, 2, 3, 4] def even(x): ? ? return x % 2 == 0
方法1: 創(chuàng)建新列表,過濾元素
1a) 列表推導,創(chuàng)建新列表
使用列表推導創(chuàng)建一個新的列表,只包含你不想刪除的元素,并把它分配回a:
a = [1, 2, 2, 3, 4] def even(x): ? ? return x % 2 == 0 ? # 列表推導,但創(chuàng)建了一個新的變量a a = [x for x in a if not even(x)] # --> a = [1, 3] print(a)
1b) 列表推導,對a[:]賦值
上面的代碼創(chuàng)建了一個新的變量a。我們也可以通過賦值給切片a[:]就地改變現(xiàn)有的列表。這種方法更有效率,如果有其他對a的引用需要反映變化的話,這種方法可能很有用。
a = [1, 2, 2, 3, 4] def even(x): ? ? return x % 2 == 0 ? # 列表推導,但賦值給a[:] 就地改變列表 a[:] = [x for x in a if not even(x)] # --> a = [1, 3] print(a)
1c) 使用itertools.filterfalse()
itertools模塊為非常有效的循環(huán)迭代提供了各種函數(shù),并且提供了一種過濾元素的方法。
a = [1, 2, 2, 3, 4] def even(x): ? ? return x % 2 == 0 ? # 通過itertools 快速過濾 from itertools import filterfalse a[:] = filterfalse(even, a) # --> a = [1, 3] print(a)
方法2:列表副本上迭代
如果你真的想保留for語法,那么需要在列表的副本上進行迭代(副本可以通過使用a[:]簡單創(chuàng)建)。
現(xiàn)在你可以在條件為True時從原始列表中刪除元素:
a = [1, 2, 2, 3, 4] def even(x): ? ? return x % 2 == 0 ? # 注意是在列表副本a[:] 上循環(huán) for item in a[:]: ? ? if even(item): ? ? ? ? a.remove(item) # --> a = [1, 3] print(a)
常見陷阱
千萬別在同一個列表上循環(huán),并在迭代過程中修改它!
這和上面的代碼是一樣的,只是沒有在副本上循環(huán)。刪除一個元素將使所有后續(xù)元素向左移動一個位置,因此在下一次迭代中,一個元素將被跳過。
這可能會導致不正確的結(jié)果:
a = [1, 2, 2, 3, 4] def even(x): ? ? return x % 2 == 0 ? # 直接在變量a上進行循環(huán),沒有在副本上 for item in a: ? ? if even(item): ? ? ? ? a.remove(item) # --> a = [1, 2, 3] !!! print(a)
另外,在列表的循環(huán)過程中,千萬不要修改索引!
這是不正確的,因為在循環(huán)中改變i不會影響下一次迭代中i的值。這個例子也會產(chǎn)生非預期的效果,甚至會導致IndexErrors,
比如這里:
a = [1, 2, 2, 3, 4] def even(x): ? ? return x % 2 == 0 ? # 試圖在循環(huán)在改變索引i,但出錯! for i in range(len(a)): ? ? if even(a[i]): ? ? ? ? del a[i] ? ? ? ? i -= 1 # --> IndexError: list index out of range print(a)
到此這篇關于python 安全地刪除列表元素的方法的文章就介紹到這了,更多相關python 刪除列表元素內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家!
相關文章
python讀取與處理netcdf數(shù)據(jù)方式
今天小編就為大家分享一篇python讀取與處理netcdf數(shù)據(jù)方式,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-02-02
python中閉包Closure函數(shù)作為返回值的方法示例
閉包(closure)是函數(shù)式編程的重要的語法結(jié)構(gòu),Python也支持這一特性,下面這篇文章主要給大家介紹了關于python中閉包Closure函數(shù)作為返回值的相關資料,文中通過示例代碼介紹的非常詳細,需要的朋友可以參考下。2017-12-12
Python中自定義函方法與參數(shù)具有默認值的函數(shù)
這篇文章主要介紹了Python中自定義函方法與參數(shù)具有默認值的函數(shù),在Python編程中,可以使用已經(jīng)定義好的函數(shù),也可以自定義函數(shù)實現(xiàn)某些特殊的功能,更多相關資料,請需要的人參考下面文章內(nèi)容2022-02-02

