numpy.diagflat

numpy.diagflat(v, k=0)

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

参数：

v：array_like

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

k：int，可选

对角线设置； 0，默认，对应于“主”对角线，一个正（负）k给出对角线上方（下方）的对数。

返回：

out：ndarray

2-D输出数组。

也可以看看

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