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

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

用法:

csr_matrix.diagonal(k=0)#

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

参数 ：：

k： 整数，可选

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

例子：

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