将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
。我成像有很好的理由来总结重复,但是,这不是我需要的。
答案 0 :(得分:0)
试试这段代码,
coo_mat.todia().tocsr()