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

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)

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

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

  b.append(int(input()))

b.append(list(map(int,input().split())))

a=[]
n=int(input())
s=0
for i in range(n):
  b=[]
  for j in range(n):  
    q=int(input())
    b.append(q)
  a.append(b)
for i in a:
    print(i,end='')
    print()
for i in range(n):
  for j in range(n):
    if i==j:
      s+=a[i][j]
print(s)

a = list()
for i in range(int(input())):
    a.append(list(map(int, input().split())))
... # тут считаешь сумму элементов главной диагонали

n = int(input())
a = [[int(x) for x in input().split()] for _ in range(n)]
print(sum([a[i][i] for i in range(n)]))

(lambda a: print(sum([a[i][i] for i in range(len(a))])))([[int(x) for x in input().split()] for _ in range(int(input()))])