NumPy - 選擇一個範圍內唯一整數的隨機列表

Question

在n = range(1, n+1)以內我必須隨機生成一個n * p唯一整數列表。

例如，如果n = 10和p = 0.3那麼可能的結果可能是：

[2, 6, 9] 
[1, 5, 7] 
[3, 4, 8] 
[3, 5, 6] 
etc 

下面的Python代碼做這項工作完美：

import random 

n = 250000 
p = 0.8 
np = int(n * p) 

result = [] 
for i in range(np): 
    attempt = random.randint(1, n) 
    if attempt not in result: 
     result.append(attempt) 

然而，因爲它是Python的，它可能需要很長（如超過一分鐘）爲np > 200000。

您是否可以使用NumPy查看上述解決方案的更高效版本？

Answer 1

嘗試這樣的：

result = random.sample(range(1,n+1),p*n) 

鏈接到documentation of random.sample

Answer 2

製作result一個集合而不是一個列表將使您無需在每次迭代時遍歷列表中的每個項目，以檢查您要附加的新項目是否已存在。小改進，但你應該看到性能的差異。