scipy.sparse.dok_matrix.

trace

dok_matrix.trace(offset=0)

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

参数：:

offsetint, 可选

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