SciPy

scipy.sparse.csr_matrix.setdiag

csr_matrix.setdiag(values, k=0)[source]

Set diagonal or off-diagonal elements of the array.

Parameters
valuesarray_like

New values of the diagonal elements.

Values may have any length. If the diagonal is longer than values, then the remaining diagonal entries will not be set. If values are longer than the diagonal, then the remaining values are ignored.

If a scalar value is given, all of the diagonal is set to it.

kint, optional

Which off-diagonal to set, corresponding to elements a[i,i+k]. Default: 0 (the main diagonal).

Previous topic

scipy.sparse.csr_matrix.set_shape

Next topic

scipy.sparse.csr_matrix.sign