Python numpy.diagflat函数方法的使用

NumPy（Numerical Python的缩写）是一个开源的Python科学计算库。使用NumPy，就可以很自然地使用数组和矩阵。NumPy包含很多实用的数学函数，涵盖线性代数运算、傅里叶变换和随机数生成等功能。本文主要介绍一下NumPy中diagflat方法的使用。

numpy.diagflat

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

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

参数：

v ：array_like

输入数据被展平并设置为输出的第k个对角线。

k ：int, 可选

对角线设置；默认值0对应于 “main” 对角线，

正（负）k表示主线以上（以下）的对角线编号。

返回值：

out ：ndarray

二维输出array。

例子

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