Matrice avec comprehension

La matrice identité est un outil fondamental en algèbre linéaire, en data science et en 3D. Savoir la générer avec une compréhension imbriquée, c'est maîtriser un pattern que tu réutiliseras pour créer des grilles, des tableaux de distances, ou toute structure 2D.

La diagonale, c'est là où l'indice de ligne égale celui de colonne (i == j). Une compréhension imbriquée utilise deux boucles for, et l'expression ternaire (1 if i == j else 0) choisit la valeur.

Visuellement, une matrice identité 3x3 :
[[1, 0, 0],
[0, 1, 0],
[0, 0, 1]]

Écris une fonction matrice_identite(n) qui retourne une matrice identité n x n en utilisant une compréhension imbriquée.

Exemple :
matrice_identite(3) renvoie [[1,0,0],[0,1,0],[0,0,1]]

Tests (2/4)

3x3

assert matrice_identite(3) == [[1,0,0],[0,1,0],[0,0,1]]

1x1

assert matrice_identite(1) == [[1]]

+ 0 tests cachés

Indices (3 disponibles)

Solution officielle

def matrice_identite(n):
    return [[1 if i == j else 0 for j in range(n)] for i in range(n)]

← Carres pairs Comprehension avec zip →

solution.py

Bravo!

Matrice avec comprehension

Tests (2/4)

Indices (3 disponibles)

Solution officielle