我在表單中有一個大列表（1e8+ 個條目）[index:boolean]。我想找到真值的索引。在這項任務(wù)中的表現(xiàn)是最重要的。目前，從我在 Python 3.7.2 中可以看出，執(zhí)行此操作的最佳方法是使用列表理解，如下所示：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 秒）。我曾希望這個操作會快得多。
1 回答
胡子哥哥
TA貢獻1825條經(jīng)驗 獲得超6個贊
在這項任務(wù)中的表現(xiàn)是最重要的
那么你應(yīng)該考慮使用一個 numpy 數(shù)組：
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幾秒鐘。
添加回答
舉報
0/150
提交
取消