Генератор чисел. Задача на Python

Question

«ГНЧЭ-1» — сложное устройство, выдающее каждую секунду очередное число последовательности 1, 2, 2, 3, 3, 3, 4, 4, 4, 4, 5... 
Вам поручено разработать эмулятор для этого устройства. 
Формат входных данных 
Дано целое количество секунд 𝑠 (1≤𝑠≤106), которые работает генератор после включения. 
Формат выходных данных 
Требуется вывести результат работы генератора.

enin_arseniy · Accepted Answer

Такая задача у меня была на плюсах...
 s = int(input()) &#10; &#10;result = [] &#10;number = 1 &#10;count = 0 &#10; &#10;while count < s: &#10;    for _ in range(number): &#10;        if count < s: &#10;            result.append(number) &#10;            count += 1 &#10;    number += 1 &#10; &#10;print(' '.join(map(str, result)))

metamorphosa · Answer

print([i for i in range(1, int(input())+1) for _ in range(i)])