在numpy中,可以使用np.diag()
函数来创建一个ndarray的对角线视图。该函数可以接受一个一维数组作为参数,将其作为对角线元素创建一个二维数组。同时,也可以通过指定偏移量来创建非主对角线的视图。
下面是对np.diag()
函数的详细解释:
概念:
np.diag()
函数是numpy库中的一个函数,用于创建一个二维数组,其中对角线元素由输入的一维数组确定。
分类:
np.diag()
函数属于numpy库中的数组操作函数。
优势:
np.diag()
函数提供了一种简单的方式来创建对角线视图,无需手动遍历数组元素。应用场景:
推荐的腾讯云相关产品和产品介绍链接地址:
代码示例:
import numpy as np
# 创建一个一维数组
arr = np.array([1, 2, 3])
# 创建对角线视图
diag_view = np.diag(arr)
print(diag_view)
输出结果:
[[1 0 0]
[0 2 0]
[0 0 3]]
以上代码中,我们首先创建了一个一维数组arr
,然后使用np.diag()
函数将其转换为对角线视图diag_view
。最后,我们打印出了对角线视图的结果。
领取专属 10元无门槛券
手把手带您无忧上云