Problem

8 /12


Réseau diagonal

Theory Click to read/hide

Matrices diagonales

La fonction  diag(V, k=0) vous permet d'extraire une diagonale d'un tableau, ainsi que de construire des tableaux diagonaux à partir de tableaux unidimensionnels.
V - Un objet de type tableau, des tableaux bidimensionnels ou unidimensionnels, des matrices, des listes ou des tuples, ou toute fonction ou objet avec une méthode qui renvoie une liste ou un tuple.
k - indice de la diagonale (facultatif).
La valeur par défaut est k = 0 qui correspond à la diagonale principale. Une valeur k positive déplace la diagonale vers le haut, une valeur négative la déplace vers le bas.

La fonction renvoie un tableau NumPy (ndarray) - la diagonale du tableau spécifié ou un tableau diagonal du tableau unidimensionnel spécifié.

Problem

L'entrée est le nombre n. Générez un tableau de taille nxn, dans dans les nombres en diagonale vont de 0 à n-1, et les autres nombres sont 0 .
 

 

Exemples
# Entrée Sortie
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]]