Вектор из максимальных элементов диагоналей целочисленного квадратного массива

from random import randint
def foo(l):
    v = []
    for i in range(1, n):
        tt = tb = []
        for j in range(n - i):
            tt.append(m[j][j + i])
            tb.append(m[j + i][j])
        v.append(max(tt))
        v.append(max(tb))
    return v
n = 5
m = [[randint(10, 99) for _ in range(n)] for _ in range(n)]
print(*m, sep='\n')
print(foo(m))

from random import randint
 
 
def f(a):
    result = []
 
    for i in range(len(a) - 1, -1, -1):
        j, tmp = 0, []
        for k in range(i, len(a)):
            tmp.append(a[k][j])
            j += 1
        result.append(max(tmp))
 
    for j in range(1, len(a)):
        i, tmp = 0, []
        for k in range(j, len(a)):
            tmp.append(a[i][k])
            i += 1
        result.append(max(tmp))
 
    return result
 
 
n = 5
m = [[randint(10, 99) for _ in range(n)] for _ in range(n)]
print(*m, sep='\n')
 
print(f(m))

import numpy as np
 
n = 7
m = np.random.randint(-10, 10, (n, n))
print(m, end='\n\n')
 
print(np.vectorize(lambda x: np.max(np.diag(m, k=x)))(np.delete(np.arange(1 - n, n), n - 1)))

np.vectorize(lambda x: np.diag(m, x).max())(np.arange(1-n,n))