突然記起來一年前剛入行時,領導面試我時提的問題,刪除list中所有等於某個值的元素
比如 刪除 [1,2,3,4,1,2,3,4,1,3,4] 中所有的2元素
刪除元素常用 remove, pop 方法,
remove可以直接刪除某個值:
list.remove(1)
但是每次只能刪除第一個,且如果list中不存在這個值則會拋出異常。
pop可以根據索引來刪除:
list.pop(2)
但是同樣在此處不適用,因爲 pop 都是直接作用於list對象,
當我們在遍歷list中,直接刪除會對遍歷有影響。
當時我給出的方案是建立一個新list,遍歷原list,如果不等於這個值,則放入新list中。。。
所以落下一個 雖然不會,但是敢想 的評價。。
當時事後回想,覺得應該是有兩種方案:
1.死循環無限remove,遇到異常則return
2.從後向前遍歷pop,遇到異常return
但是現在想起來,完全可以有更好的方法:
filter函數
def remove_item(n):
return n != 1
aaa = [1,2,3,5,6,8,9,5,3,1]
print list(filter(remove_item, aaa))
同樣,用於去除list中的空字符串或者None同樣好用:
new_list = list(filter(None, list))