我正在尝试以区间[-5,-1]和[1,5]的方式随机创建矩阵。如何创建一个考虑到具有随机值的间隔(无0值)的矩阵?
我尝试使用numpy的randint
和piecewise
来做到这一点。在第一种情况下,不可能指示2个间隔,在第二种情况下,当值接近0(大于-1且小于1)时生成的随机数始终是相同的。
x=np.random.randint(-5,5,(2,3))
np.piecewise(x, [x <= 1, x >=-1 ], [np.random.randint(-5,-1), np.random.randint(1,5)])
我想以这种方式创建一个没有0值的矩阵:
[-2,-3,-1
1,-2, 3]
答案 0 :(得分:6)
您可以创建一个随机的正矩阵,并将其按元素乘以一个随机符号:
np.random.randint(1,6,(2,3))*np.random.choice([-1,1],(2,3))
请记住,这仅在间隔大约为0
时才有效,例如(-6,6)
(排除0
)。
答案 1 :(得分:5)
一种选择是简单地将值0“重新映射”到其中一个边界:
import numpy as np
np.random.seed(100)
x = np.random.randint(-5, 5, (4, 8))
x[x == 0] = 5
print(x)
# [[ 3 3 -2 2 2 -5 -1 -3]
# [ 5 -3 -3 -3 -4 -5 3 -1]
# [-5 4 1 -3 -1 -4 5 -2]
# [-1 -1 -2 2 -4 -4 2 2]]
或者您可以使用np.random.choice
:
import numpy as np
np.random.seed(100)
np.random.choice(list(range(-5, 0)) + list(range(1, 6)), size=(4, 8))
print(x)
# [[ 3 3 -2 2 2 -5 -1 -3]
# [ 5 -3 -3 -3 -4 -5 3 -1]
# [-5 1 -3 -1 -4 5 -2 -1]
# [-1 -2 2 -4 -4 2 2 -5]]
答案 2 :(得分:2)
np.random.choice(list(range(-5,0))+list(range(1,6)),(5,5))
或
a = list(range(-5,6))
a.remove(0)
np.random.choice(a,(5,5))
答案 3 :(得分:0)
import numpy as np
a = np.arange(-5,6)
np.random.choice(list(numpy.select([a != 0, a == 0], [a, -1])),(5,5))
这是定义列表条件的另一种方法。您可以通过编辑条件来省略数组'a'中所需的任何数字。在上面的数字中,等于零的数字返回-1,对于等于零的数字,返回相同的数字。