numpy.diag() en Python

numpy.diag(a, k=0) : Extrae y construye una array diagonal 
Parámetros : 

a : array_like 
k : [int, optional, 0 by default]
          Diagonal we require; k>0 means diagonal above main diagonal or vice versa.

Devoluciones : 

ndarray

# Python Programming illustrating
# numpy.diag method
  
import numpy as geek
  
# matrix creation by array input
a = geek.matrix([[1, 21, 30], 
                 [63 ,434, 3], 
                 [54, 54, 56]])
  
print("Main Diagonal elements : \n", geek.diag(a), "\n")
  
print("Diagonal above main diagonal : \n", geek.diag(a, 1), "\n")
  
print("Diagonal below main diagonal : \n", geek.diag(a, -1))

Producción : 
 

Main Diagonal elements : 
 [  1 434  56] 

Diagonal above main diagonal : 
 [21  3] 

Diagonal below main diagonal : 
 [63 54]

Referencias: 
https://docs.scipy.org/doc/numpy/reference/generated/numpy.diagflat.html#numpy.diagflat  
Nota: 
estos programas NumPy-Python no se ejecutarán en IDE en línea, así que ejecútelos en sus sistemas para explorarlos 
. 
Este artículo es aportado por Mohit Gupta_OMG 😀 . Si te gusta GeeksforGeeks y te gustaría contribuir, también puedes escribir un artículo usando write.geeksforgeeks.org o enviar tu artículo por correo a review-team@geeksforgeeks.org. Vea su artículo que aparece en la página principal de GeeksforGeeks y ayude a otros Geeks.
Escriba comentarios si encuentra algo incorrecto o si desea compartir más información sobre el tema tratado anteriormente.