numpy.diag() 函数返回一个新的 NumPy 数组，该数组包含输入数组对角线上的元素
numpy.diag()
import numpy as np a = np.array([[1, 2], [3, 4]]) result = np.diag(a) print(result) # 输出：[1 4]
在这个例子中，np.diag() 函数从二维数组 a 中提取了对角线上的元素（1 和 4），并将它们存储在一个新的一维数组 result 中。
np.diag()
a
result
