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

Question

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

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

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

我得到了我想要的正确结构和密度，但填充它的值是介于 0 和 1 之间的 float 。

有没有办法用 0 或 1 的 float 生成这种结构？

Answer 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。