python – 生成仅填充值0或1的随机稀疏矩阵

2023年4月24日 436次阅读

我正在尝试使用SciPy生成随机csr_matrix,但我需要它只填充值0或1.

到目前为止我正在尝试使用：

rand(1000,10,density = 0.2,format =’csr’,random_state = np.random.randint(0,2))

我得到了我想要的正确结构和密度,但填充它的值是0到1之间的浮点数.

有没有办法只用0或1的浮点数生成这个结构？

最佳答案 您可以简单地用随机矩阵替换随机矩阵中的非零值：

from scipy.sparse import rand

x = rand(1000, 10, density=0.2, format='csr')
x.data[:] = 1

print(np.unique(x.todense().flat))
# [ 0.  1.]

我不认为random_state = kwarg做你认为它做的事情 – 它只是允许你为随机数生成器指定种子,或者显式传递np.random.RandomState实例作为RNG.

点赞