Problem

8 /12


matriz diagonal

Theory Click to read/hide

Matrices diagonales

La función  diag(V, k=0) le permite extraer una diagonal de una matriz, así como crear matrices diagonales a partir de matrices unidimensionales.
V - Un objeto similar a un arreglo, arreglos bidimensionales o unidimensionales, matrices, listas o tuplas, o cualquier función u objeto con un método que devuelve una lista o tupla.
k - índice de la diagonal (opcional).
El valor predeterminado es k = 0 que corresponde a la diagonal principal. Un valor k positivo mueve la diagonal hacia arriba, un valor negativo la mueve hacia abajo.

La función devuelve array NumPy (ndarray) - la diagonal de matriz especificada o una matriz diagonal de la matriz unidimensional especificada.

Problem

La entrada es el número n. Genere una matriz de tamaño nxn, in in los números diagonales son 0 a n-1, y el resto de los números son 0 .
 

 

Ejemplos

# Entrada Salida
1 5 [[0 0 0 0 0]
 [0 1 0 0 0]
 [0 0 2 0 0]
 [0 0 0 3 0]
 [0 0 0 0 4]]