画鸵萌宠网
np.diag() 是 NumPy 库中的一个函数,用于创建一个对角矩阵或从一个矩阵中提取对角线元素。具体来说,np.diag() 可以接受以下两种参数:
除此之外,np.diag() 还可以接受一个整数参数 k,表示对角线偏移量。当 k>0 时,将提取矩阵的主对角线之上的元素,k<0 时,则提取主对角线之下的元素。例如,np.diag([[1,2,3],[4,5,6],[7,8,9]], k=1) 将返回一个一维数组,包含矩阵 [[1,2,3],[4,5,6],[7,8,9]] 主对角线之上的元素 [2,6]。
总之,np.diag() 是一个十分实用的函数,可以用于创建对角矩阵或提取矩阵的对角线元素。
因篇幅问题不能全部显示,请点此查看更多更全内容