Python-сообщество

 matrix[i][j] = int(input()).split()

 # задаем размерность, пустой список и счетчик для матрицы
rows = int(input())
matrix = []
count = 0
# создаем вложенный список для будущей матрицы заполненный нулями
for _ in range(rows):
    matrix.append([0] * rows)
# заполняем матрицу произвольными значениями и сразу вычисляем сумму чисел главной диагонали
for i in range(rows):
    for j in range(rows):
        matrix[i][j] = int(input())
        if (i == j):                       # это индексы элементов, а не значения
            count += matrix[i][j]   # а здесь уже значения соответствующие этим индексам
# выводим матрицу на экран
for r in range(rows):
    for c in range(rows):
        print(str(matrix[r][c]).rjust(3), end=' ')
    print()
# принтуем сумму чисел главной диагонали
print(count)