Найти сумму элементов над и под главной и побочной диагоналями

from random import randint
N = int(input())
lst=[[randint(1, 9) for i in range(N)] for i in range(N)]
for i in lst:
    print()
    for j in i:
        print (j, end=" ")
sum=0
for i in range(1, N):
    for j in range(i):
        sum = sum+lst[i][j]
print('\nсумма под главной диагональю', sum)
sum1=0
for i in range(i):
    for j in range(j):
        if i<N-j-1:
            sum1 = sum1+lst[i][j]
print('\nсумма выше побочной', sum1)
sum2=0
for i in range(i):
    for j in range(j):
        if i>N-j-1:
            sum2 = sum2+lst[i][j]
print('\nсумма ниже побочной', sum2)

from random import randint
 
N = int(input())
lst = [[randint(1, 9) for _ in range(N)] for _ in range(N)]
for i in lst:
    print(*i)
summ = 0
for i in range(1, N):
    summ += sum(lst[i][:i])
print('\nсумма под главной диагональю', summ)
sum1 = 0
for i in range(N):
    sum1 += sum(lst[i][:N-1-i])
print('\nсумма выше побочной', sum1)
sum2 = 0
for i in range(1, N):
    sum2 += sum(lst[i][-i:])
print('\nсумма ниже побочной', sum2)

from random import randint
 
N = int(input())
lst = [[randint(1, 9) for _ in range(N)] for _ in range(N)]
for i in lst:
    print(*i)
print(f'\nсумма под главной диагональю: {sum(sum(lst[i][:i]) for i in range(1, N))}')
print(f'\nсумма выше побочной: {sum(sum(lst[i][:N - 1 - i]) for i in range(N))}')
print(f'\nсумма ниже побочной: {sum(sum(lst[i][-i:]) for i in range(1, N))}')

import numpy as np
n = 3 
 
ar = np.random.randint(0, 9, size=(n,n))
print('Исходная матрица:\n', ar)
print('Сумма элементов под главной диагональю:')
print(sum(np.sum([[j for j in np.diag(ar, k = -i)] for i in range(1, n)])) )
ar1 = np.rot90(ar)
print('Сумма элементов под побочной диагональю:')
print(sum(np.sum([[j for j in np.diag(ar1, k = i)] for i in range(1, n)])) )