最佳列表理解（过滤现有列表）

Question

我在表单中有一个大列表（1e8+ 个条目）[index:boolean]。我想找到真值的索引。在这项任务中的表现是最重要的。目前，从我在 Python 3.7.2 中可以看出，执行此操作的最佳方法是使用列表理解，如下所示：return [i for i, j in enumerate(numbers) if j]我还尝试了以下方法（尽管它似乎只是早期 Python 版本的首选方法）：return list(filter(lambda a: a, numbers))第二种方法比第一种方法慢约 25%。目前，此操作大约需要 (0.8*x) 时间，而我算法的实际逻辑部分需要“x”时间。（例如，如果逻辑需要 10 秒，从列表中提取正值需要大约 8 秒）。我曾希望这个操作会快得多。

Answer 1

在这项任务中的表现是最重要的

那么你应该考虑使用一个 numpy 数组：

import numpy as np

from random import choice

from timeit import Timer

bools = True, False

li = [choice(bools) for _ in range(int(1e8))]

arr = np.array(li)  

print(Timer(lambda: np.nonzero(arr)).repeat(1, 1))

输出

[0.4524359999999916]

那是0.4524359999999916几秒钟。