numpy.diagflat#

numpy.diagflat(v, k=0)[源代码]#

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

参数:

varray_like: 输入数据,它被展平并设置为输出的第 k 个对角线.
kint, 可选: 对角线设置;0,默认值,对应于”主”对角线,正（负）`k`表示主对角线上方（下方）的对角线编号.

返回:

outndarray: 二维输出数组.

参见

diag: MATLAB 的 1-D 和 2-D 数组类似物.
diagonal: 返回指定的对角线.
trace: 沿对角线求和.

示例

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