官方文档

功能

提取一个对角数组或构造一个对角数数组（矩阵）。
当传入的量为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。

使用案例