numpy

numpy

  numpy的random用于生成随机数。

randint

  randint用于生成随机整数：

import numpy as np

# 生成[0, 9)之间的随机整数矩阵，大小为(3, 3)
a = np.random.randint(9, size=(3, 3))
print("a =", a)
# 生成5个[0, 2)之间的整数
b = np.random.randint(0, 2, 5)
print("b =", b)

执行结果：

a = [[3 7 1]
 [1 2 8]
 [2 4 7]]
b = [1 0 1 0 0]