Вложенные списки

a=[]
n=int(input())
for i in range(n):
  b=[]
  for j in range(n):
    b.append(int(input()))
  a.append(b)
for i in a:
  print(i)

n = int(input())
a = []
for _ in range(n):
    *r, = map(int, input().split())
    a.append(r)
s = 0
for i, r in enumerate(a):
    s += r[i]
print(s)

a = []
n = int(input())
for i in range(n):
    a.append(list(map(int, input().split())))
for i in a:
    print(i)

n = int(input())
result = []
for i in range(n):
    temp = input().split()
    line = [int(temp[0]), int(temp[1])]
    result.append(line)
 
summ = 0
for line in result:
    for elem in line:
        print(elem, end=' ')
        summ += elem
    print()
 
print(summ)

n = int(input())
a = []
for _ in range(n):
    a.append(list(map(int, input().split())))
s = 0
for i in range(len(a)):
    s += a[i][i]
print(s)

n = int(input()) (вводим число и преобразуем его в целочисленный тип данных)
a = [] (создаем пустой список)
for _ in range(n): (цикл от 0 до n - 1)
    a.append(list(map(int, input().split()))) (ввод чисел в одну строку с разделителем в виде пробела)
s = 0  (объявляем и присваиваем значение 0)
for i in range(len(a)): (цикл от 0 до (размера а - 1))
    s += a[i][i] (к значению s прибавляем значение диагонального элемента)
print(s)(вывод)