scipy.sparse.dia_array.

trace

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

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

Параметры:

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

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