python - 一致地创建相同的随机 numpy 数组

Question

我正在等待另一位开发人员完成一段代码，该代码将返回一个形状为 (100,2000) 的 np 数组，其值为 -1,0 或 1。

与此同时，我想随机创建一个具有相同特征的数组，以便在开发和测试中抢占先机。问题是我希望这个随机创建的数组每次都相同，这样我就不会针对每次重新运行进程时都会不断更改其值的数组进行测试。

我可以像这样创建我的数组，但是有没有办法创建它，以便每次都一样。我可以腌制对象并取消腌制，但想知道是否还有其他方法。

r = np.random.randint(3, size=(100, 2000)) - 1

Answer 1

使用您选择的种子创建您自己的 numpy.random.RandomState() 实例。请勿使用 numpy.random.seed()，除非解决不允许您传递自己的 RandomState 实例的不灵活库。

[~]
|1> from numpy.random import RandomState

[~]
|2> prng = RandomState(1234567890)

[~]
|3> prng.randint(-1, 2, size=10)
array([ 1,  1, -1,  0,  0, -1,  1,  0, -1, -1])

[~]
|4> prng2 = RandomState(1234567890)

[~]
|5> prng2.randint(-1, 2, size=10)
array([ 1,  1, -1,  0,  0, -1,  1,  0, -1, -1])