Python SciPy dia_matrix.diagonal用法及代碼示例

本文簡要介紹 python 語言中 scipy.sparse.dia_matrix.diagonal 的用法。

用法:

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