numpy.diagflat

numpy.diagflat(v, k=0)

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

参数

varray_like

输入数据，扁平化并设置为 k -输出的第对角线。

k可选的

要设置的对角线；默认为0，对应于“主”对角线，为正（负） k 给出主对角线上方（下方）的编号。

返回

out恩达雷

二维输出数组。

参见

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

numpy.diag numpy.tri