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