numpy.diagflat

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

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

Параметры

varray_like

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

kint, optional

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

Возвращает

outndarray

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

См. также

diag

Аналог MATLAB для 1-D и 2-D массивов.

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