scipy.sparse.csr_array.

diagonal

csr_array.diagonal(k=0)

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

参数:

kint, 可选

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

在版本 1.0 中添加。

另请参阅

numpy.diagonal

等效的 NumPy 函数。

示例

在您的浏览器中试用！

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

返回在新标签页中打开