python - 我可以生成一个包含负数和正数的数组并指定这个数组的限制和维度吗？

Question

c=np.random.rand(10,2)从 [0,1) 生成随机数数组.我可以生成相同的数组(就大小而言)但带有负数和正数吗？此外，我可以选择这个数组的限制和维度吗？例如，如果我想要从 -2 到 2。

Answer 1

使用 np.random.randint 并传递 size 参数。对于 [-2, 2) 和大小 (10, 2) 中的值，您有:

In [32]: np.random.randint(-2, 2, (10, 2))
Out[32]:
array([[-1,  1],
       [-2,  0],
       [-1,  0],
       [-2, -2],
       [ 0, -2],
       [-2, -2],
       [ 0,  0],
       [ 1, -1],
       [ 0,  1],
       [-2, -1]])

或者对浮点数使用 np.random.uniform:

In [33]: size = (10, 2)

In [39]: np.random.uniform(-2, 2, size)
Out[39]:
array([[-1.1129566 , -1.94562947],
       [ 0.21356557, -1.96769933],
       [ 1.1481992 , -0.08494563],
       [ 0.12561175,  0.95580417],
       [-1.79335536, -1.01276994],
       [ 1.56808971,  0.15911181],
       [ 0.50987451, -1.39039728],
       [-1.57962641,  1.59555514],
       [-1.8318709 , -1.57055885],
       [ 0.682527  , -1.89022731]])