稀疏矩阵的相似度计算

Question

我使用带有numpy，scipy和scikit-learn模块的Python。

我想把数组分类成非常大的稀疏矩阵。(100,000 * 100,000)

矩阵中的值等于0或1。我唯一有的就是值= 1的索引。

a = [1,3,5,7,9] 
b = [2,4,6,8,10]

这意味着

a = [0,1,0,1,0,1,0,1,0,1,0]
b = [0,0,1,0,1,0,1,0,1,0,1]

如何在scipy中将索引数组更改为稀疏数组？

如何快速对这些数组进行分类？

非常感谢。

Answer 1

如果您选择稀疏coo_matrix，则可以通过如下索引创建它：

from scipy.sparse import coo_matrix
import scipy
nrows = 100000
ncols = 100000
row = scipy.array([1,3,5,7,9])
col = scipy.array([2,4,6,8,10])
values = scipy.ones(col.size)
m = coo_matrix((values, (row,col)), shape=(nrows, ncols), dtype=float)