如何使用 numpy 創建一個非重復隨機數矩陣，同時在這些隨機數之間有一個范圍？

Question

這行代碼已經創建了一個 3x3 矩陣，其中填充了 0 到 8 之間的隨機數，但其中一些是重復的。 如何阻止這種情況？

initStateRandom = np.random.randint(9, size=(3,3))

Answer 1

您可以使用來自np.random的choice function，請參閱文檔。 您可以為范圍或任何 ndarray 傳入一個 int。

import numpy as np

r_array = np.random.choice(9, size=(3, 3), replace=False)  # int
r_array = np.random.choice(np.arange(10, 100), size=(3, 3), replace=False)  # ndarray

Answer 2

您也可以使用“np.random.shuffle”。 它就地工作，並且僅沿多維 arrays 的一維工作，因此您需要像這樣洗牌一個臨時的一維數組：

    import numpy as np

    tmp=np.arange(9)
    np.random.shuffle(tmp)
    mat=np.reshape(tmp,[3,3])