scipy.sparse.coo_array.

trace

coo_array.trace(offset=0)

返回稀疏数组/矩阵对角线上的和。

参数:

偏移量int, 可选

获取哪条对角线，对应于元素 a[i, i+offset]。默认值：0（主对角线）。