numpy.diagflat
numpy.diagflat(v, k=0) [source]
创建一个二维数组，将扁平输入作为对角线。
参数 ： | v ：array_like 输入数据被展平并设置为输出的第k个对角线。 k ： 对角线设置； 默认值0对应于 “main” 对角线， 正（负）k表示主线以上（以下）的对角线编号。 |
返回值 ： | out ：ndarray 二维输出array。 |
例子
import numpy as np
# 示例 1: 创建一个对角矩阵，主对角线上是输入数组的元素
arr = [1, 2, 3, 4]
diag_matrix = np.diagflat(arr)
print("对角矩阵：")
print(diag_matrix)
# 示例 2: 通过指定 k 参数，创建一个偏移的对角矩阵
arr = [1, 2, 3, 4]
diag_matrix_k = np.diagflat(arr, k=1)
print("\n偏移对角矩阵：")
print(diag_matrix_k)
# 示例 3: 输入二维数组
arr_2d = np.array([[1, 2], [3, 4]])
diag_matrix_2d = np.diagflat(arr_2d)
print("\n二维数组生成的对角矩阵：")
print(diag_matrix_2d)