如何避免在将COO转换为CSR矩阵时对重复项进行求和

时间:2016-10-18 00:27:08

标签: python matrix

将COO矩阵转换为CSR格式时,重复项将相加。在scipy的earler版本中有一个标志sum_duplicates=False,但现在它已经消失了。有没有办法将COO转换为CSR矩阵而不总结重复项?

以下是生成coo矩阵的示例:

from scipy.sparse import csr_matrix,coo_matrix

i_tar = np.array([3,3,3,3,3,6,7,8,8,2,2,2])
j_src = np.array([8,8,8,8,8,5,7,18,18,4,4,4])
data = np.array([3,4,6,8,40,6,12,5,6,2,3,9])

coo_mat = coo_matrix((data, (i_tar,j_src)),shape=(20,20))

print coo_mat
  (3, 8)  3
  (3, 8)  4
  (3, 8)  6
  (3, 8)  8
  (3, 8)  40
  (6, 5)  6
  (7, 7)  12
  (8, 18) 5
  (8, 18) 6
  (2, 4)  2
  (2, 4)  3
  (2, 4)  9

将其转换为我们发现的CSR格式:

print coo_mat.tocsr()
  (2, 4)  14
  (3, 8)  61
  (6, 5)  6
  (7, 7)  12
  (8, 18) 11
总结了重复的

。我成像有很好的理由来总结重复,但是,这不是我需要的。

1 个答案:

答案 0 :(得分:0)

试试这段代码,

coo_mat.todia().tocsr()