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

发布时间：2020-12-20 11:42:21 所属栏目：Python 来源：网络整理

导读：我正在尝试使用SciPy生成随机csr_matrix,但我需要它只填充值0或1. 到目前为止我正在尝试使用： rand(1000,10,density = 0.2,format =’csr’,random_state = np.random.randint(0,2)) 我得到了我想要的正确结构和密度,但填充它的值是0到1之间的浮点数. 有没

我正在尝试使用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,density=0.2,format='csr')
x.data[:] = 1

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

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

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!