基于一维列表从二维列表中删除元素

Question

我有 2 个列表

列表 A（称为 a）：像这样的 2D 列表：[[1,'aaa'],[2,'bbb'],[3,'ccc'],[4,'ddd']]

列表 B（称为 b）：像这样的一维列表：['aaa','abc','cba','acb']

我想根据列表 B 的内容从列表 A 中删除元素。 在示例中，预期结果是：

新建列表C(叫c): [[2,'bbb'],[3,'ccc'],[4,'ddd']] //既然没有元素'bbb', 'ccc'或列表中的“ddd”B

我目前正在使用以下代码，我发现它非常慢：

c = []
for elem in a:
    if elem[1] not in b:
        c.append(elem)

有没有更好的方法来做这个删除？创建一个新列表并添加元素或从原始列表中删除元素更好吗？

感谢您的帮助！

Answer 1

没有快多少，但使用列表理解可能更干净

c = [elem for elem in a if elem[1] not in b]

---

如果b 非常大，那么转换为集合将显着加快速度，因为列表中的查找是线性时间O(n)，但集合中的查找是恒定时间O(1)

b_set = set(b)
c = [elem for elem in a if elem[1] not in b_set]

---

如果a（'aaa', 'bbb', 等）的 index-1 元素是唯一的，那么这可能会更快，因为那时我们可以使用 set difference operator -

a_dict = {k: v for v, k in a}
b_set = set(b)

set_difference = a_dict.keys() - b_set
c = [(a_dict[k]: k) for k in set_difference]