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]])