scipy.sparse.coo_matrix.

trace

coo_matrix.trace(смещение=0)

Возвращает сумму по диагоналям разреженного массива/матрицы.

Параметры:

смещениеint, необязательный

Какую диагональ получить, соответствующую элементам a[i, i+offset]. По умолчанию: 0 (главная диагональ).