numpy函数：[17]diag函数

numpy.diag()返回一个矩阵的对角线元素，或者创建一个对角阵（ diagonal array.）。我们还是用例子来说明问题：

先引入numpy



创建一个对角阵，如下



创建一个二维数组x



提起二维数组的对角元素的值，构成一个一位数组



或者，我们可以使用第二个参数，获取其他元素，请自行观察这个值在x数组中的位置



接着，你换一个参数试试



连续使用两个diag，可以得到一个对角阵，除对角线以外的元素均为零



今天用到的所有代码：

>>> numpy.diag((1,2,3))

array([[1, 0, 0],

       [0, 2, 0],

       [0, 0, 3]])

>>> 

>>> 

>>> x=numpy.arange(9).reshape((3,3))

>>> x

array([[0, 1, 2],

       [3, 4, 5],

       [6, 7, 8]])

>>> 

>>> 

>>> x=numpy.arange(10,19).reshape((3,3))

>>> x

array([[10, 11, 12],

       [13, 14, 15],

       [16, 17, 18]])

>>> 

>>> 

>>> y=numpy.diag(x)

>>> y

array([10, 14, 18])

>>> 

>>> 

>>> 

>>> numpy.diag(x,1)

array([11, 15])

>>> 

>>> numpy.diag(x,-1)

array([13, 17])

>>> 

>>> 

>>> numpy.diag(np.diag(x))

Traceback (most recent call last):

  File "<pyshell#38>", line 1, in <module>

    numpy.diag(np.diag(x))

NameError: name 'np' is not defined

>>> numpy.diag(numpy.diag(x))

array([[10,  0,  0],

       [ 0, 14,  0],

       [ 0,  0, 18]])

>>>