在numpy中，我如何從一個範圍中隨機選擇一個數字，排除之前選擇的隨機數字？

Question

例如：

open_door = np.random.randint(1,4) 
prize_behind_door = np.random.randint(1,4) 
change_door = np.random.randint(1,4) 

但我不想change_door有一個值是一樣的open_door，而它仍然是隨機的（所以它可以等於prize_behind_door），什麼命令我應該使用？

Answer 1

使用np.setdiff1d設置排除open_door一個範圍，然後用np.random.choice選擇關閉它一個隨機元素 -

np.random.choice(np.setdiff1d(range(1,4), open_door)) 

樣品試驗 -

In [329]: open_door = np.random.randint(1,4) 
    ...: prize_behind_door = np.random.randint(1,4) 
    ...: 

In [330]: open_door 
Out[330]: 2 

In [331]: prize_behind_door 
Out[331]: 1 

In [349]: np.setdiff1d(range(1,4), open_door) # array that excludes open_door 
Out[349]: array([1, 3]) 

In [332]: np.random.choice(np.setdiff1d(range(1,4), open_door)) 
Out[332]: 1 

In [333]: np.random.choice(np.setdiff1d(range(1,4), open_door)) 
Out[333]: 3 

In [334]: np.random.choice(np.setdiff1d(range(1,4), open_door)) 
Out[334]: 3 

In [335]: np.random.choice(np.setdiff1d(range(1,4), open_door)) 
Out[335]: 3 

In [336]: np.random.choice(np.setdiff1d(range(1,4), open_door)) 
Out[336]: 1 

Answer 2

如何choice沒有replace？

In [230]: np.random.choice(range(1,4),2, replace=False) 
Out[230]: array([2, 3]) 
In [231]: np.random.choice(range(1,4),2, replace=False) 
Out[231]: array([2, 3]) 
In [232]: np.random.choice(range(1,4),2, replace=False) 
Out[232]: array([1, 3]) 
In [233]: np.random.choice(range(1,4),2, replace=False) 
Out[233]: array([1, 2]) 
In [234]: np.random.choice(range(1,4),2, replace=False) 
Out[234]: array([3, 2]) 
In [235]: np.random.choice(range(1,4),2, replace=False) 
Out[235]: array([3, 2]) 
In [236]: open_door, change_door = np.random.choice(range(1,4),2, replace=False) 
    ...: 
In [237]: open_door 
Out[237]: 3 
In [238]: change_door 
Out[238]: 1 

2值將是隨機的，但總是不同。這個選擇將獨立於prize_behind_door。

對於從一個小範圍的這樣的2個值，Python的random模塊將是一樣好，如果沒有更好的（快的時間測試表明它的速度更快）：

random.sample(range(1,4),2) 

Answer 3

我會用while循環

open_door = np.random.randint(1,4) 
prize_behind_door = np.random.randint(1,4) 
while True: 
    x = np.random.randint(1,4) 
    if x != open_door: 
    change_door = x 
    break 

如果你想disclude多個值，你可以只添加if語句

if x != value_1 and x != value_2: 
    value_3 = x 
    break 

另一種方法是創建一個長度與隨機數範圍相同的布爾數組。

如果布爾型數組的索引值爲False，則表示該值尚未被選中，如果它爲true，則表示它已經存在。

在while循環中檢查是否通過獲取bool_array [chosen_number]的結果來選擇值。如果它已被忽略並選擇一個新的數字，否則將該值附加到您用於表示值的數組，並將該數字的索引的布爾值數組設置爲True以表示它現在被選中。

我會做這樣一個函數內保持一切井井有條

from random import randint  

def select_random_numbers(rand_range, values): 

    temp_list = [] 
    isPicked = [False for _ in range(rand_range)] 

    for i in range(values): 
    while True: 
     x = randint(0, rand_range - 1) 
     if isPicked[x] == False: 
     temp_list.append(x) 
     isPicked[x] = True 
     break 
    return(temp_list) 


print(select_random_numbers(10, 5))