Matriz de identidad

Dada una dimensión n, genera una matriz de identidad de tamaño n × n.

Una matriz de identidad es una matriz cuadrada donde:

• Los elementos de la diagonal principal (donde el índice de fila = índice de columna) son 1.
• Todos los demás elementos son 0.

Ejemplo

identity_matrix(3)
# [
#   [1, 0, 0],
#   [0, 1, 0],
#   [0, 0, 1]
# ]

identity_matrix(1)
# [[1]]

Restricciones

• 1 ≤ n ≤ 100
• La función debe retornar una lista de listas de enteros.