過濾兩份名單同時

Question

我有三個列表：

del_ids = [2, 4] 
ids = [3, 2, 4, 1] 
other = ['a', 'b', 'c', 'd'] 

，我的目標是消除del_ids，結果是

ids = [3, 1] 
other = ['a', 'd'] 

我試圖做一個面具元素，以保持（mask = [id not in del_ids for id in ids] ），我打算在這兩個列表上應用此掩碼。

但我覺得這不是一個pythonic解決方案。你能告訴我怎樣才能做得更好嗎？

Answer 1

拉鍊，過濾並再次解壓縮：

ids, other = zip(*((id, other) for id, other in zip(ids, other) if id not in del_ids)) 

的zip()呼叫對每個id與相應other元件，發電機表達濾除其中id在del_ids被列出的任何一對，並且zip(*..)然後引子再次將剩餘的對放入單獨的列表中。

演示：

>>> del_ids = [2, 4] 
>>> ids = [3, 2, 4, 1] 
>>> other = ['a', 'b', 'c', 'd'] 
>>> zip(*((id, other) for id, other in zip(ids, other) if id not in del_ids)) 
[(3, 1), ('a', 'd')] 

Answer 2

壓縮，過濾器，解壓：

ids, other = zip(*filter(lambda (id,_): not id in del_ids, zip(ids, other)))