ret[np.diff(csr_mat.indptr) != 0] = np.maximum.reduceat(csr_mat.data,
    csr_mat.indptr[:-1][np.diff(csr_mat.indptr) > 0])
return ret
эффективный способ получить максимум каждой строки для большой разреженной матрицы