Python3列表：如果el [0]不在list_b中，如何刪除list_a中的el？

Question

假設 A = [（1， 'CA'），（2，「NY） B = [2,3,4,5,6]

如何刪除EL在埃爾若[0]不在b中？在這種情況下，我想刪除（1，'CA'），因爲1不在b上。我嘗試在list_a中使用el，但由於列表不斷變化，所以無法使用。 謝謝！

Answer 1

由於您將首先重複檢查元素是否在b,convert b to a set之間，因此每次檢查是O(1)而不是O(n)。然後你可以使用list comprehension：

[el for el in a if el[0] in bset] 

---

a = [(1, 'CA'), (2, 'NY')] 
b = [2,3,4,5,6] 
bset = set(b) 

a = [el for el in a if el[0] in bset] 
print(a) 

產量

[(2, 'NY')] 

---

注：如果a是非常短的，不轉換成一組實際速度是：

In [4]: %timeit bset = set(b); [el for el in a if el[0] in bset] 
1000000 loops, best of 3: 580 ns per loop 

In [5]: %timeit [el for el in a if el[0] in b] 
1000000 loops, best of 3: 335 ns per loop 

但是，如果a長，然後轉換b一組是值得的：

In [47]: a = a*1000 

In [48]: %timeit bset = set(b); [el for el in a if el[0] in bset] 
10000 loops, best of 3: 157 us per loop 

In [49]: %timeit [el for el in a if el[0] in b] 
10 loops, best of 3: 37.2 ms per loop 

Answer 2

這裏是一個一行。 bset=set(b)只執行一次

>>> a = [(1, 'CA'), (2, 'NY')] 
>>> b = [2, 3, 4, 5, 6] 
>>> a[:] = filter(lambda x, bset=set(b): x[0] in bset, a) 
>>> a 
[(2, 'NY')]