scipy.sparse.dia_matrix.

setdiag

dia_matrix.setdiag(values, k=0)

設定陣列/矩陣的對角線或非對角線元素。

參數:

values類陣列

對角線元素的新值。

值可以是任何長度。如果對角線比值長，則剩餘的對角線條目將不會被設定。如果值比對角線長，則剩餘的值將被忽略。

如果給定純量值，則所有對角線都將設定為該值。

k整數，選填

要設定哪個非對角線，對應於元素 a[i,i+k]。預設值：0（主對角線）。