scipy.sparse.coo_matrix.

setdiag

coo_matrix.setdiag(values, k=0)

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

參數:

values類陣列

對角線元素的新值。

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

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

k整數，選用

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