Numpy中np.random.randint怎么使用

小億
123
2024-02-04 10:15:47

np.random.randint函數(shù)用于產(chǎn)生指定范圍內(nèi)的隨機(jī)整數(shù)。

函數(shù)的語(yǔ)法為: np.random.randint(low, high=None, size=None, dtype=int)

參數(shù)說(shuō)明:

  • low:生成的隨機(jī)數(shù)的最小值(包含)。
  • high:生成的隨機(jī)數(shù)的最大值(不包含)。如果不指定該參數(shù),則隨機(jī)數(shù)的最大值為low。
  • size:生成隨機(jī)數(shù)的個(gè)數(shù)或形狀。如果指定為整數(shù),則生成一個(gè)具有該個(gè)數(shù)元素的一維數(shù)組;如果指定為元組或列表,則生成具有該形狀的數(shù)組。默認(rèn)為None,表示生成一個(gè)隨機(jī)數(shù)。
  • dtype:生成隨機(jī)數(shù)的數(shù)據(jù)類型,默認(rèn)為整型。

下面是一些使用np.random.randint函數(shù)的例子:

  1. 生成一個(gè)隨機(jī)整數(shù): import numpy as np

rand_num = np.random.randint(10) print(rand_num)

輸出:一個(gè)0-9之間的隨機(jī)整數(shù)

  1. 生成一個(gè)指定范圍內(nèi)的隨機(jī)整數(shù): rand_num = np.random.randint(1, 10) print(rand_num)

輸出:一個(gè)1-9之間的隨機(jī)整數(shù)

  1. 生成一個(gè)指定范圍內(nèi)的多個(gè)隨機(jī)整數(shù): rand_arr = np.random.randint(1, 10, size=5) print(rand_arr)

輸出:一個(gè)包含5個(gè)1-9之間的隨機(jī)整數(shù)的一維數(shù)組

  1. 生成一個(gè)指定形狀的隨機(jī)整數(shù)數(shù)組: rand_arr = np.random.randint(1, 10, size=(2, 3)) print(rand_arr)

輸出:一個(gè)2行3列的隨機(jī)整數(shù)數(shù)組,數(shù)組元素為1-9之間的隨機(jī)整數(shù)

這些只是np.random.randint函數(shù)的一些常見用法,根據(jù)具體需求可以靈活調(diào)整參數(shù)。

0