如何使用 python 以更快的方式生成序列号列表

Question

任何人都可以提高处理时间以生成序列号列表？ 这是我的代码，它需要 ~ 0.05 秒。

import torch
import time
import random
index = [torch.tensor(660000)]
st = time.time()
allowed = [x for x in range(index[0])]
index = random.sample(allowed, 1000)
print(time.time()-st)

请指教

谢谢你

Answer 1

这应该有助于 OP 特定情况，而不是问题标题中所述的“生成序列号列表”。

您不需要创建整个列表，您可以提供范围，无论范围大小如何，这都将以相同的速度工作。

index = random.sample(range(index[0]), 1000)

在我的机器上，范围大小为 100 万的速度要快约 100 倍

---

基于这个答案