Python numpy.diagflat()

numpy。Diagflat (a, k = 0):创建一个二维数组，使用array_like输入作为新输出数组的对角线。

参数 :

**a :** 类似数组的输入数据，有对角线元素
strong>k :[int, optional, 默认为0]
          我们需要的对角线；k>0意味着对角线高于主对角线，反之亦然。

返回值 :

数组，以array_like的输入作为对角线到新的输出阵列。

# Python Program illustrating
# numpy.diagflat method
 
import numpy as geek
 
print("diagflat use on main diagonal : \n", geek.diagflat([1, 7]), "\n")
 
print("diagflat use on main diagonal : \n", geek.diagflat([1, 7, 6]), "\n")
 
# Diagonal above main diagonal
print("diagflat above main diagonal : \n", geek.diagflat([1, 7, 6], 1), "\n")

输出 :

diagflat use on main diagonal  : 
 [[1 0]
 [0 7]] 

diagflat use on main diagonal  : 
 [[1 0 0]
 [0 7 0]
 [0 0 6]] 

diagflat above main diagonal  : 
 [[0 1 0 0]
 [0 0 7 0]
 [0 0 0 6]
 [0 0 0 0]]