python - 获取大型稀疏矩阵每行最大值的有效方法

Question

我有一个很大的稀疏矩阵，我想获得每一行的最大值。在 numpy 中，我可以调用 numpy.max(mat, axis=1)，但我找不到 scipy 稀疏矩阵的类似函数。有什么有效的方法可以获取大型稀疏矩阵的每一行的最大值吗？

Answer 1

如果您的矩阵，我们称它为 a，以 CSR 格式存储，则 a.data 具有按行排序的所有非零条目，并且 a.indptr 具有每行第一个元素的索引。您可以使用它来计算您想要的内容，如下所示:

def sparse_max_row(csr_mat):
    ret = np.maximum.reduceat(csr_mat.data, csr_mat.indptr[:-1])
    ret[np.diff(csr_mat.indptr) == 0] = 0
    return ret