np.random.randint函數用于產生指定范圍內的隨機整數。
函數的語法為:
np.random.randint(low, high=None, size=None, dtype=int)
參數說明:
- low:生成的隨機數的最小值(包含)。
- high:生成的隨機數的最大值(不包含)。如果不指定該參數,則隨機數的最大值為low。
- size:生成隨機數的個數或形狀。如果指定為整數,則生成一個具有該個數元素的一維數組;如果指定為元組或列表,則生成具有該形狀的數組。默認為None,表示生成一個隨機數。
- dtype:生成隨機數的數據類型,默認為整型。
下面是一些使用np.random.randint函數的例子:
- 生成一個隨機整數:
import numpy as np
rand_num = np.random.randint(10)
print(rand_num)
輸出:一個0-9之間的隨機整數
- 生成一個指定范圍內的隨機整數:
rand_num = np.random.randint(1, 10)
print(rand_num)
輸出:一個1-9之間的隨機整數
- 生成一個指定范圍內的多個隨機整數:
rand_arr = np.random.randint(1, 10, size=5)
print(rand_arr)
輸出:一個包含5個1-9之間的隨機整數的一維數組
- 生成一個指定形狀的隨機整數數組:
rand_arr = np.random.randint(1, 10, size=(2, 3))
print(rand_arr)
輸出:一個2行3列的隨機整數數組,數組元素為1-9之間的隨機整數
這些只是np.random.randint函數的一些常見用法,根據具體需求可以靈活調整參數。