numpy.diagflat#

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

创建一个二维数组，其对角线由扁平化的输入构成。

参数:

varray_like: 输入数据，将被扁平化并作为输出数组的第 k 条对角线。
kint，可选: 要设置的对角线；默认值 0 对应于“主”对角线，正数 (负数) k 表示主对角线之上 (之下) 的对角线序号。

返回:

outndarray: 二维输出数组。

另请参阅

diag: 适用于一维和二维数组的 MATLAB 类似功能。
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]])