pyramid.py

def pyramid(n: int) -> list:
    size = 2 * n - 1
    table = [size * [0] for _ in range(size)]
    for start in range(n):
        for j in range(size - start):
            table[start][start + j] = start + 1
        for j in range(size - start):
            table[size - 1][start + j] = start + 1
        for i in range(size - start):
            table[start + i][start] = start + 1
        for i in range(size - start):
            table[start + i][size - 1] = start + 1
        size -= 1
    return table


if __name__ == "__main__":
    size = int(input("size = "))
    for row in pyramid(n=size):
        for element in row:
            print(element, end=" ")
        print()