📜  Python中的numpy.diagflat

📅  最后修改于: 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]]