我在表單中有一個(gè)大列表（1e8+ 個(gè)條目）[index:boolean]。我想找到真值的索引。在這項(xiàng)任務(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) 時(shí)間，而我算法的實(shí)際邏輯部分需要“x”時(shí)間。（例如，如果邏輯需要 10 秒，從列表中提取正值需要大約 8 秒）。我曾希望這個(gè)操作會(huì)快得多。
1 回答
胡子哥哥
TA貢獻(xiàn)1825條經(jīng)驗(yàn) 獲得超6個(gè)贊
在這項(xiàng)任務(wù)中的表現(xiàn)是最重要的
那么你應(yīng)該考慮使用一個(gè) 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幾秒鐘。
添加回答
舉報(bào)
0/150
提交
取消