np.random.randint()介绍

np.random.randint() 是 NumPy 库中的一个随机整数生成函数，用于生成指定范围内的随机整数。它可以用于生成随机索引、随机种子和在一定范围内的随机整数等操作。

函数语法

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

参数说明

• low：表示生成的随机整数的最小值。如果没有给出 high 值，则0为最小值
• high：表示生成的随机整数的最大值，不包括在生成的范围内。如果没有给出 high 值，则其默认为生成的最大值。
• size：表示要生成的随机整数的形状，可以用一个整数或一个整数的元组进行表示。如果没有给出，则为一个随机整数。
• dtype：生成的随机数的数据类型，默认为 int。
• generator：随机数生成器，可以是 None，即默认生成器。

返回值

函数将会返回一个符合给定shape的随机整数数组。

示例代码

import numpy as np

# 生成0-5范围内的随机整数，生成1x5数组
a = np.random.randint(6, size=5)
print(a)

# 生成0-5范围内的随机整数，并将其整形为2x4数组
b = np.random.randint(6, size=(2, 4))
print(b)

# 生成1-9范围内的随机整数，并将其整形为2x3数组
c = np.random.randint(1, 10, size=(2, 3))
print(c)

# 生成3范围内的随机整数，并将其整形为2x2x2数组
d = np.random.randint(3, size=(2, 2, 2))
print(d)

以上示例代码将会输出如下结果：

[0 5 1 1 0]
[[3 1 1 3]
 [5 2 5 5]]
[[7 7 1]
 [5 6 1]]
[[[0 0]
  [0 2]]

 [[1 2]
  [2 2]]]

注意事项

• low 必须小于等于 high。如果 high 被忽略，则默认其取值为 low。
• high 值本身不会在生成的整数范围中出现。
• dtype 默认为 int，但可以是其他整数类型。但请注意，这将在数组中保留更小的值范围。由于可用值小于 int 类型，因此可能会发生溢出错误。