numpy.ma.diagflat#

ma.diagflat = <numpy.ma.extras._fromnxfunction_single object>#

创建一个二维数组,将扁平化的输入作为对角线。

参数:
varray_like

输入数据,它会被扁平化并设置为输出的第 k 个对角线。

kint, 可选

要设置的对角线;0 是默认值,对应于“主”对角线,正(负)k 表示主对角线之上(之下)的对角线编号。

返回:
outndarray

二维输出数组。

参见

diag

适用于一维和二维数组的 MATLAB 类比。

diagonal

返回指定对角线。

trace

沿对角线求和。

备注

该函数同时应用于 _data 和 _mask(如果有)。

示例

>>> import numpy as np
>>> 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]])