Faça um programa para retornar os valores de uma matriz 3x3 de números inteiros. Depois da leitura dos dados - Lista de Exercícios Resolvidos de Python

Valor para a linha 0 e coluna 0: 1
Valor para a linha 0 e coluna 1: 4
Valor para a linha 0 e coluna 2: 7
Valor para a linha 1 e coluna 0: 12
Valor para a linha 1 e coluna 1: 9
Valor para a linha 1 e coluna 2: 8
Valor para a linha 2 e coluna 0: 5
Valor para a linha 2 e coluna 1: 10
Valor para a linha 2 e coluna 2: 14

Valores na matriz

    1     4     7 
   12     9     8 
    5    10    14 

A soma dos elementos da diagonal principal é: 24

# método principal
def main():
  # uma matriz de três linhas e três colunas
  matriz = [[0 for x in range(3)] for y in range(3)]
  soma_diagonal = 0 # guarda a soma dos elementos na diagonal principal
  
  # vamos ler os valores para os elementos da matriz
  for i in range(len(matriz)):
    for j in range(len(matriz[i])):
      matriz[i][j] = int(input("Elemento na linha [%d] e coluna [%d]: " % (i, j)))
   
  # vamos mostrar a matriz da forma que ela foi informada
  print("\nValores na matriz:\n");
     
  for i in range(len(matriz)):
    for j in range(len(matriz[i])):
      print("%5d  " % matriz[i][j], end=' ')
    print()
   
  # vamos calcular a soma dos elementos da diagonal principal
  for i in range(len(matriz)):
    for j in range(len(matriz[i])):
      if(i == j):
        soma_diagonal = soma_diagonal + matriz[i][j]
   
  print("\nA soma dos elementos da diagonal principal é: {0}".format(soma_diagonal)) 
  
if __name__== "__main__":
  main()