scipy.sparse.csr_array.

setdiag

csr_array.setdiag(values, k=0)

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

参数:

valuesarray_like

对角线元素的新值。

值可以具有任意长度。如果对角线比值长，则不会设置剩余的对角线条目。如果值比对角线长，则忽略剩余的值。

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

kint, 可选

要设置哪个非对角线，对应于元素 a[i,i+k]。默认值：0 (主对角线)。