python - 将稀疏矩阵一分为二

Question

问题：如何根据列表中的值将 1 个稀疏矩阵拆分为 2 个？

也就是说，我有一个稀疏矩阵X：

>>print type(X)
<class 'scipy.sparse.csr.csr_matrix'>

我在脑海中将其想象为列表列表，如下所示：

>>print X.todense()
[[1,3,4]
 [3,2,2]
 [4,8,1]]

我有一个y看起来像这样的列表：

y = [-1, 
      3, 
     -4]

如何X根据对应的值y是正数还是负数将其分成两个稀疏矩阵？例如，我怎样才能得到：

>>print X_pos.todense()
 [[3,2,2]] 
>>print X_neg.todense()
 [[1,3,4]
  [4,8,1]]

结果 (X_pos和X_neg) 显然也应该是稀疏矩阵，因为它只是分割一个稀疏矩阵开始。

谢谢！

Answer 1

用于np.where为正值和负值生成两个索引数组y，然后使用它们来索引到您的稀疏矩阵。

>>> X = csr_matrix([[1,3,4], [3,2,2], [4,8,1]])
>>> y = np.array([-1, 3, -4])
>>> y_pos = np.where(y > 0)[0]
>>> y_neg = np.where(y < 0)[0]
>>> X_pos = X[y_pos]
>>> X_neg = X[y_neg]

您现在必须包含所需元素的 CSR 矩阵：

>>> X_pos
<1x3 sparse matrix of type '<type 'numpy.int64'>'
    with 3 stored elements in Compressed Sparse Row format>
>>> X_neg
<2x3 sparse matrix of type '<type 'numpy.int64'>'
    with 6 stored elements in Compressed Sparse Row format>
>>> X_pos.A
array([[3, 2, 2]])
>>> X_neg.A
array([[1, 3, 4],
       [4, 8, 1]])