numpy.diagflat

numpy.diagflat(v, k=0) [source]

Создать двумерный массив с уплощенным входом в качестве диагонали.

Параметры

varray_like

Входные данные, которые уплощаются и устанавливаются в качестве k-й диагонали выходных данных.

kint, optional

Диагональ для установки; 0, по умолчанию, соответствует «главной» диагонали, положительное (отрицательное) k значение дает номер диагонали выше (ниже) главной.

Возвращает

outndarray

Двумерный выходной массив.

См. также

diag

Аналог MATLAB для одномерных и двумерных массивов.

diagonal

Возвратить указанные диагонали.

trace

Сумма по диагоналям.

Примеры

>>> np.diagflat([[1,2], [3,4]])
array([[1, 0, 0, 0],
       [0, 2, 0, 0],
       [0, 0, 3, 0],
       [0, 0, 0, 4]])

>>> np.diagflat([1,2], 1)
array([[0, 1, 0],
       [0, 0, 2],
       [0, 0, 0]])