Является ли scipy.sparse.csr_matrix.sum (axis = None) слишком сложным? - PullRequest
0 голосов
/ 27 февраля 2019

csr_matrix и csc_matrix имеют все свои элементы в поле .data, поэтому не проще и (НАМНОГО) быстрее вернуть его сумму, когда axis is None?

Мои тесты показали, чтодля 99% разреженной матрицы из 2M элементов csr_matrix.sum() занимает ~ 1,2 мс, а data.sum() занимает ~ 14 мкс, и результаты всегда проходят np.isclose тест.

Исходный код

...