📅  最后修改于: 2020-06-04 14:50:25             🧑  作者: Mango
numpy.diagflat (a, k = 0) : 创建一个二维数组,其array_like输入作为新输出数组的对角线。
参数:
a:具有对角线元素的array_like输入数据
strong> k:[int,可选,默认为0]
我们需要对角线; k> 0表示对角线高于主对角线,反之亦然。
返回值:
array_like输入作为新输出数组的对角线的数组。
# Python程序说明numpy.diagflat方法
import numpy as geek
print("在主对角线上使用diagflat : \n", geek.diagflat([1, 7]), "\n")
print("在主对角线上使用diagflat : \n", geek.diagflat([1, 7, 6]), "\n")
print("在主对角线上使用diagflat : \n", geek.diagflat([1, 7, 6], 1), "\n")
输出:
在主对角线上使用diagflat:
[[1 0]
[0 7]]
在主对角线上使用diagflat:
[[1 0 0]
[0 7 0]
[0 0 6]]
在主对角线上使用diagflat:
[[0 1 0 0 ]
[0 0 7 0]
[0 0 0 6]
[0 0 0 0]]