a： array_like 
k： [int，可选，默认为0] 
          我们需要对角线；k> 0表示对角线高于主对角线，反之亦然。

# Python编程说明numpy.diag方法 
  
import numpy as geek 
  
# 通过数组输入创建矩阵 
a = geek.matrix([[1, 21, 30],  
                 [63 ,434, 3],  
                 [54, 54, 56]]) 
  
print("Main Diagnol elements : \n", geek.diag(a), "\n") 
  
print("Diagnol above main diagnol : \n", geek.diag(a, 1), "\n") 
  
print("Diagnol below main diagnol : \n", geek.diag(a, -1))

主要诊断元素：
 [1 434 56] 

在主要诊断之上的诊断：
 [21 3] 

在主要诊断之下的诊断：
 [63 54]