Exercício Resolvido de Python NumPy - Como gerar uma matriz identidade de ordem N usando Python e NumPy - Exercícios Resolvidos de Python

Informe a ordem da matriz: 4
A matriz identidade gerada foi:

[[1. 0. 0. 0.]
 [0. 1. 0. 0.]
 [0. 0. 1. 0.]
 [0. 0. 0. 1.]]

# vamos importar a biblioteca NumPy
import numpy as np

# função principal do programa
def main():
  # vamos ler a ordem da matriz
  ordem = int(input("Informe a ordem da matriz: "))

  # vamos gerar a matriz identidade
  matriz_identidade = np.eye(ordem)

  # vamos exibir o resultado
  print("A matriz identidade gerada foi:\n")
  print(matriz_identidade)

if __name__== "__main__":
  main()