Problem

8 /12


matrice diagonale

Theory Click to read/hide

Array diagonali

La  diag(V, k=0) funzione consente di estrarre una diagonale da un array, nonché di creare array diagonali da array unidimensionali.
V - Un oggetto simile a un array, array bidimensionali o unidimensionali, matrici, elenchi o tuple o qualsiasi funzione o oggetto con un metodo che restituisce un elenco o una tupla.
k - indice della diagonale (facoltativo).
Il valore predefinito è k = 0 che corrisponde alla diagonale principale. Un valore k positivo sposta la diagonale verso l'alto, un valore negativo verso il basso.

La funzione restituisce array NumPy (ndarray) - la diagonale dell'array specificata o una matrice diagonale dall'array unidimensionale specificato.

Problem

L'input è il numero n. Produce un array di dimensione nxn, in i diagonali i numeri vanno da 0 a n-1, e il resto dei numeri è 0 .
 

 

Esempi
# Input Uscita
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]]