Массив: Вывести матрицу, у которой каждый элемент равен сумме элементов матрицы на заданных позициях... - Python - Ответ 13464077

lst = []
st = ''
sub = 0
out_list = []
while st != 'end':
    st = input()
    if st != 'end':
        st = st.split(' ')
        a = []
        for z in st: 
            a.append(int(z))
        lst.append(a)
i = 0
j = 0
for i in range(len(lst)):
    row_lst = []
    for j in range(len(lst[i])):
        if j+1 < len(lst[i]) and i+1 < len(lst):
            sub = lst[i][j-1] + lst[i][j+1] + lst[i-1][j] + lst[i+1][j]
            sub = str(sub)
            row_lst.append(sub)
        else:
            sub = lst[i][j-1] + lst[i][0] + lst[i-1][j] + lst[0][j]
            sub = str(sub)
            row_lst.append(sub)
    print(' '.join(row_lst))

lst = []
st = ''
sub = 0
out_list = []
while st != 'end':
    st = input()
    if st != 'end':
        st = st.split(' ')
        a = []
        for z in st: 
            a.append(int(z))
        lst.append(a)
for i in range(len(lst)):
    row_lst = []
    for j in range(len(lst[i])):
        if j+1 < len(lst[i]) and i+1 < len(lst):
            sub = lst[i][j-1] + lst[i][j+1] + lst[i-1][j] + lst[i+1][j]
            sub = str(sub)
            row_lst.append(sub)
        elif j+1 == len(lst[i]) and i+1 < len(lst):
            sub = lst[i][j-1] + lst[i][0] + lst[i-1][j] + lst[i][j]
            sub = str(sub)
            row_lst.append(sub)
        elif j+1 < len(lst[i]) and i+1 == len(lst):
            sub = lst[i][j-1] + lst[i][j+1] + lst[i-1][j] + lst[0][j]
            sub = str(sub)
            row_lst.append(sub)
        else:
            sub = lst[i][j-1] + lst[i][0] + lst[i-1][j] + lst[0][j]
            sub = str(sub)
            row_lst.append(sub)
    print(' '.join(row_lst))