scipy.sparse.dia_array.

setdiag

dia_array.setdiag(values, k=0)

设置数组/矩阵的对角线或非对角线元素。

参数：:

**values**array_like

对角线元素的新值。

值可以是任何长度。如果对角线比值长，则剩余的对角线条目将不会被设置。如果值比对角线长，则剩余的值将被忽略。

如果给定一个标量值，则所有对角线都将被设置为该值。

**k**int, 可选

要设置的非对角线，对应于元素 a[i,i+k]。默认值：0（主对角线）。