Python numpy diagflat用法及代码示例

本文简要介绍 python 语言中 numpy.diagflat 的用法。

用法:

numpy.diagflat(v, k=0)

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

参数：

v： array_like

输入数据，被展平并设置为输出的k-th 对角线。

k： 整数，可选

对角线设置； 0，默认值，对应于 “main” 对角线，正(负)k 给出主上(下)对角线的编号。

返回：

out： ndarray

二维输出数组。

例子：

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