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 中。
亿速云「云服务器」，即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘，价格低至29元/月。点击查看>>
相关推荐：python中函数返回值是什么