python diag函数的返回值是什么

numpy.diag() 函数返回一个新的 NumPy 数组，该数组包含输入数组对角线上的元素

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 中。