Вывести матрицу того же размера, у которой каждый элемент в позиции i, j равен сумме элементов первой матрицы - Python - Ответ 6923305

data = []
while True:
    inp = input()
    if inp == 'end':
        break
    data.append(inp.split())
 
p = len(data[0])
k = ' '
if len(data) == 1:
    print(int(data[0][0]) * 4)
else:
    for i in range(0, len(data)):
        for j in range(0, p):
            if j == p - 1:
                k = '\n'
            if i == 0 and j == 0:
                print(int(data[i][j + 1]) + int(data[i + 1][j]) + int(data[i][p - 1]) + int(data[len(data) - 1][j]), end = k)
            elif i == 0 and j == p - 1:
                print(int(data[i][j - 1]) + int(data[i + 1][j]) + int(data[i][0]) + int(data[len(data) - 1][j]), end = k)
            elif i == len(data) - 1 and j == 0:
                print(int(data[i][j + 1]) + int(data[i - 1][j]) + int(data[i][p - 1]) + int(data[0][j]), end = k)
            elif i != 0 and j == 0:
                print(int(data[i][j + 1]) + int(data[i + 1][j]) + int(data[i][p - 1]) + int(data[i - 1][j]), end = k)
            elif i == 0 and j != 0:
                print(int(data[i][j - 1]) + int(data[i + 1][j]) + int(data[i][j + 1]) + int(data[len(data) - 1][j]), end = k)
            elif i == len(data) - 1 and j == p - 1:
                print(int(data[i][j - 1]) + int(data[i - 1][j]) + int(data[i][0]) + int(data[0][j]), end = k)
            elif i != 0 and j == p - 1:
                print(int(data[i][j - 1]) + int(data[i - 1][j]) + int(data[i][0]) + int(data[i + 1][j]), end = k)
            elif i == len(data) - 1 and j != 0:
                print(int(data[i][j - 1]) + int(data[i - 1][j]) + int(data[i][j + 1]) + int(data[0][j]), end = k)
            else:
                print(int(data[i - 1][j]) + int(data[i][j + 1]) + int(data[i + 1][j]) + int(data[i][j - 1]), end = k)
            k = ' '

string = m = []
while string != 'end':
    string = input()
    m.append(list(map(int, string.split(' ')))) if string != 'end' else None
li, lj = len(m), len(m[0])
new = [[sum([m[i-1][j], m[(i+1)%li][j], m[i][j-1], m[i][(j+1)%lj]]) for j in range(lj)] for i in range(li)]
[print(' '.join(map(str, row))) for row in new]