Como criar uma matriz identidade em Python usando a função eye() da biblioteca NumPy

# vamos importar a biblioteca NumPy
import numpy as np

# método principal
def main():
  # vamos criar uma matriz identidade de 4 linhas e 4 colunas
  matriz_identidade = np.eye(4)

  # vamos mostrar o resultado
  print("A matriz identidade gerada foi:\n{0}".format(matriz_identidade))

if __name__== "__main__":
  main()

A matriz identidade gerada foi:
[[1. 0. 0. 0.]
 [0. 1. 0. 0.]
 [0. 0. 1. 0.]
 [0. 0. 0. 1.]]