csr_matrix
и csc_matrix
имеют все свои элементы в поле .data
, поэтому не проще и (НАМНОГО) быстрее вернуть его сумму, когда axis is None
?
Мои тесты показали, чтодля 99% разреженной матрицы из 2M элементов csr_matrix.sum()
занимает ~ 1,2 мс, а data.sum()
занимает ~ 14 мкс, и результаты всегда проходят np.isclose
тест.
Исходный код