scipy.sparse.coo_matrix.diagonal¶
- coo_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])