Numpy randint

numpy.random.randint是numpy中的一个随机数生成器函数，可返回一个指定形状和范围的随机整数数组。

语法

numpy.random.randint(low, high=None, size=None, dtype='l')

参数说明

• low：生成的随机数下限，默认为0。
• high：生成的随机数上限（不包含），默认为1。
• size：数组的形状，例如(2, 3)表示2行3列，或者3表示1行3列，默认为None，表示返回单个随机整数。
• dtype：返回数组的数据类型，默认为np.int64。

返回值

numpy.random.randint函数根据参数生成一个指定形状和范围的随机整数数组。

示例

import numpy as np

# 生成一个随机整数
num = np.random.randint(10)
print(num) # 输出一个0-9之间的随机整数

# 生成一个形状为(3,3)的随机整数数组
arr1 = np.random.randint(10, size=(3, 3))
print(arr1)

# 生成一个形状为(2,3)的随机整数数组，数据类型为np.float
arr2 = np.random.randint(1, 10, size=(2, 3), dtype=np.float)
print(arr2)

输出结果：

6
[[9 9 2]
 [7 2 8]
 [6 3 3]]
[[3. 1. 8.]
 [9. 9. 2.]]

总结

numpy.random.randint函数可以帮助程序员快速生成指定形状和范围的随机整数数组，对于一些需要随机数的场合，十分方便。