def marmottes(n):
    population = ...
    for i in ...
        ...
    return ...

print(marmottes(10))