scipy.sparse.lil_matrix.diagonal

lil_matrix.diagonal(k=0)[源代码]

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

参数
k整型,可选

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

1.0 新版功能.

参见

numpy.diagonal

等价的Numpy函数。

示例

>>> 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])