pyramide.py

def update_ring(ring_number, matrix):
    for i, row in enumerate(matrix):
        for j, value in enumerate(row):
            if value == 0 and (i in [ring_number, len(row) - ring_number - 1] or j in [ring_number, len(row) - ring_number - 1]):
                matrix[i][j] = ring_number + 1


def pyramidMatrix(N):
    matrix = []

    for i in range(N):
        matrix.append([0]*N)

    for i in range(N/2 + 1):
        update_ring(i, matrix)

    return matrix