Python numpy random.randint用法及代碼示例

本文簡要介紹 python 語言中 numpy.random.randint 的用法。

用法:

random.randint(low, high=None, size=None, dtype=int)

返回隨機整數低的(含)至高的(不包括)。

從 “half-open” 區間內指定數據類型的 “discrete uniform” 分布返回隨機整數 [低的,高的)。如果高的為 None (默認值)，則結果來自 [0,低的)。

注意

新代碼應改為使用default_rng() 實例的integers 方法；請參閱快速入門。

參數：

low： int 或類似整數的數組

要從分布中提取的最低(有符號)整數(除非high=None，在這種情況下，此參數高於最高這樣的整數)。

high： int 或類似整數的數組，可選

如果提供，則為從分布中提取的最大(有符號)整數之上的一個(請參閱上麵 high=None 的行為)。如果是類似數組，則必須包含整數值

size： int 或整數元組，可選

輸出形狀。例如，如果給定的形狀是 (m, n, k) ，則繪製 m * n * k 樣本。默認為無，在這種情況下返回單個值。

dtype： dtype，可選

結果的所需數據類型。字節序必須是原生的。默認值為 int。

返回：

out： int 或整數數組

size - 形狀的隨機整數數組，來自適當的分布，如果未提供 size，則為單個此類隨機 int。

例子：

>>> np.random.randint(2, size=10)
array([1, 0, 0, 0, 1, 1, 0, 0, 1, 0]) # random
>>> np.random.randint(1, size=10)
array([0, 0, 0, 0, 0, 0, 0, 0, 0, 0])

生成 0 到 4 之間的 2 x 4 整數數組，包括：

>>> np.random.randint(5, size=(2, 4))
array([[4, 0, 2, 1], # random
       [3, 2, 2, 0]])

生成具有 3 個不同上限的 1 x 3 數組

>>> np.random.randint(1, [3, 5, 10])
array([2, 2, 9]) # random

生成具有 3 個不同下限的 1 x 3 數組

>>> np.random.randint([1, 5, 7], 10)
array([9, 8, 7]) # random

使用 dtype 為 uint8 的廣播生成 2 x 4 數組

>>> np.random.randint([1, 3, 5, 7], [[10], [20]], dtype=np.uint8)
array([[ 8,  6,  9,  7], # random
       [ 1, 16,  9, 12]], dtype=uint8)