主页 / 问题 / 1543481
Accepted
Master's Time
Asked:2023-10-01 17:58:49 +0000 UTC2023-10-01 17:58:49 +0000 UTC

如何一次从数组中删除多个元素？

  • 772

我正在为一项工作编写算法，遇到一个问题：如何一次从数组中删除多个元素？例如，有这样的代码：

a = [1,2,3,4,5]
b = [1,3,5]
a.remove(*b)

这段代码不起作用（它给出了一个错误，.remove()它应该只获取一个元素）。

还有这样的代码：

a = [1,2,3,4,5]
b = [1,3,5]
for i in b:
    a.remove(i)

但这段代码的问题是速度很慢（并且a、 和b有超过 1000 个元素）。

有没有算法（或团队）可以解决这样的问题？

python

3 个回答

  1. Best Answer

    内置类型set（set）支持减法（-）运算：

    a = [1,2,3,4,5]
b = [1,3,5]

print(list(set(a) - set(b)))

    结论：

    [2, 4]
    • 4

  2. 元素的顺序被a保留。最小附加内存仅为b_set. a将就地更新，无需额外内存。难度是线性的。

    b_set = set(b)
a[:] = (x for x in a if x not in b_set)
    • 4

  3. 使用设定差值

    a = [1,2,3,4,5]
b = [1,3,5]

a = list(set(a) - set(b))

print(a)
    • 3

相关问题