numpy.diag

官方文档

功能

提取一个对角数组或构造一个对角数数组(矩阵)。
当传入的量为1维array,则以该array为对角线填充出一个矩阵。
当传入array为矩阵时,则输出该矩阵的对角线array.

请参阅numpy的更详细的文档。如果你使用这个函数来提取一个对角线,并希望写入结果数组;它返回的是副本还是视图取决于您使用的numpy版本。

参数

v:array_like
如果v是一个二维数组,返回它的第k个对角线的一个副本。如果v是一个1-D数组,则返回一个v在k对角线上的2-D数组。

k:int, optional
整数,可选参数,默认为0。可以简单的理解为对角线偏移量。对于主对角线以上的对角线使用k>0,对于主对角线以下的对角线使用k<0。

使用案例

image.png
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容