[转载] 生成对角矩阵 numpy.diag

参考链接： Python中的numpy.diag

给定对角线上元素，我想生成对角矩阵，在网上搜了一下，竟然都是numpy.diagonal。这个函数的作用是提取给定矩阵的对角元素，当然不是我想要的。后来发现numpy.diag才是生成对角矩阵的函数，所以写此文章记录之。 

import numpy as np

a=[1,2,3]

np.diag(a)

 

Out[4]: 

array([[1, 0, 0],

       [0, 2, 0],

       [0, 0, 3]])