python - 根据每个列表的第一个元素从列表列表中删除一个项目

Question

给定:

a = [[1,2],[3,4],[5,6],[7,8]]
b = 3

我想删除 a 中第一项为 b 的项。所以在这种情况下，我们将删除 [3,4] 以提供:

a = [[1,2],[5,6],[7,8]]

我当前的代码是:

if b in [i[0] for i in a]:
    pos = [i[0] for i in a].index(b)
       del a[pos]

这有效，但速度很慢。执行此操作的更好方法是什么？

编辑:我之前没有测试过性能，所以我可能做错了，但我明白了:

def fun1():
    lst = [[x, 2*x] for x in range(1000000)]
    lst = [x for x in lst if x[0] != 500]
    return lst

def fun2():
    lst = [[x, 2*x] for x in range(1000000)]
    for i in reversed(range(len(lst))):
        if lst[i][0] == 500:
            del lst[i]
    return lst

cProfile.runctx('fun1()', None, locals())
        6 function calls in 0.460 seconds

cProfile.runctx('fun2()', None, locals())
        6 function calls in 0.502 seconds

Answer 1

反向删除a，就地修改它:

for i in reversed(range(len(a))):
    if a[i][0] == 3:
        del a[i]

就地修改意味着效率更高，因为它不会创建新列表(就像列表理解那样)。

---

由于 OP 需要一个高性能的解决方案，这里是两个投票最高的答案之间的 timeit 比较。

设置 -

a = np.random.choice(4, (100000, 2)).tolist()

print(a[:5])
[[2, 1], [2, 2], [3, 2], [3, 3], [3, 1]]

列表理解-

%timeit [x for x in a if x[0] != b]
11.1 ms ± 685 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

反向删除-

%%timeit
for i in reversed(range(len(a))):
    if a[i][0] == 3:
        del a[i]

10.1 ms ± 146 µs per loop (mean ± std. dev. of 7 runs, 1 loop each)

它们非常接近，但是反向删除 在性能上有 1UP，因为它不必像列表理解那样在内存中生成新列表。