numpy.diagflat #

麻木的。diagflat ( v , k = 0 ) [来源] #

创建一个二维数组,其中展平的输入作为对角线。

参数
v类数组

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

k整数,可选

对角线设置;默认值 0 对应于“主”对角线,正(负)k给出主线上方(下方)的对角线的编号。

返回
输出数组

二维输出数组。

也可以看看

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