Python SciPy spmatrix.diagonal用法及代码示例

本文简要介绍 python 语言中 scipy.sparse.spmatrix.diagonal 的用法。

用法:

spmatrix.diagonal(k=0)

返回矩阵的第 k 个对角线。

参数：

k： 整数，可选

获取哪条对角线，对应元素a[i, i+k]。默认值：0(主对角线)。

例子：

>>> from scipy.sparse import csr_matrix
>>> A = csr_matrix([[1, 2, 0], [0, 0, 3], [4, 0, 5]])
>>> A.diagonal()
array([1, 0, 5])
>>> A.diagonal(k=1)
array([2, 3])