当 np.diag(array) 中

array是一个1维数组时,结果形成一个以一维数组为对角线元素的矩阵

array是一个二维矩阵时,结果输出矩阵的对角线元素

python numpy 的用法——diag函数

 

相关文章: