numpy.diagflat

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

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

Параметры:

varray_like

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

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

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

Возвращает:

outndarray

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

См. также

diag

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

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