scipy.sparse.dia_array.

setdiag

dia_array.setdiag(values, k=0)

Установить диагональные или внедиагональные элементы массива/матрицы.

Параметры:

valuesarray_like

Новые значения диагональных элементов.

Значения могут иметь любую длину. Если диагональ длиннее значений, то оставшиеся записи диагонали не будут установлены. Если значения длиннее диагонали, то оставшиеся значения игнорируются.

Если задано скалярное значение, вся диагональ устанавливается в него.

kint, необязательный

Какую побочную диагональ установить, соответствующую элементам a[i,i+k]. По умолчанию: 0 (главная диагональ).