我在表單中有一個大列表(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 秒)。我曾希望這個操作會快得多。
1 回答

胡子哥哥
TA貢獻1825條經驗 獲得超6個贊
在這項任務中的表現是最重要的
那么你應該考慮使用一個 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幾秒鐘。
添加回答
舉報
0/150
提交
取消