np.random.randint 生成 -1 +1 - Python

np.random.randint 是 NumPy 库中的一个函数，可以用来生成指定范围内的随机整数。

当给定参数为 (low, high=None, size=None, dtype=int) 时，其中的 low 表示随机整数的最小值，high 表示随机整数的最大值加一，也就是随机整数的范围是 [low, high)。如果 high 没有指定，则随机整数的范围为 [0, low)。

下面的代码演示如何用 np.random.randint 生成 $-1$ 到 $1$ 之间的随机整数：

import numpy as np

# 生成一个大小为 (5, 5) 的随机矩阵，元素的范围是 [-1, 1)
matrix = np.random.randint(-1, 2, size=(5, 5))

print(matrix)

输出：

[[ 0 -1  0  0 -1]
 [ 0  1 -1 -1 -1]
 [-1 -1  1 -1  1]
 [-1 -1  0 -1  0]
 [ 1  1  1 -1  1]]

上面的代码中，我们给定了 low=-1 和 high=2，也就是生成的随机整数会在 $-1$ 和 $1$ 之间。同时，我们把 size 设为了 (5, 5)，因此生成的是一个 $5 \times 5$ 的矩阵。