scipy.sparse.coo_matrix.

對角線

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

返回在新分頁中開啟