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]

shuffle

  shuffle用于将数组随机打乱：

import numpy as np

a = np.arange(9).reshape((3, 3))
print("a =", a)
np.random.shuffle(a)
print("a =", a)

执行结果：

a = [[0 1 2]
     [3 4 5]
     [6 7 8]]
a = [[6 7 8]
     [0 1 2]
     [3 4 5]]