scipy.sparse.csr_matrix.

setdiag

csr_matrix.setdiag(values, k=0)

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

參數:

valuesarray_like

對角線元素的新值。

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

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

kint，選擇性

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