选择如何汇总稀疏矩阵中的重复索引

时间:2019-03-01 09:46:02

标签: python scipy sparse-matrix

coo_matrix构造稀疏矩阵(例如csc_matrixcsr_matrix(data, (row_ind, col_ind)))时,默认情况下,重复(row_ind, col_ind)上的条目是总结在一起。例如:

from scipy.sparse import coo_matrix

row_ind  = [0, 0, 0, 2, 0, 0]
col_ind  = [0, 0, 0, 2, 1, 1]
data = [1, 1, 1, 1, 1, 1]

coo_matrix((data, (row_ind, col_ind))).todense()

matrix([[3, 2, 0],
        [0, 0, 0],
        [0, 0, 1]])

是否有一种方法可以指定其他操作,例如minmaxmean而不是sum?同样,对这些操作也使用类似sum_duplicates()的方法会很有用。

0 个答案:

没有答案