5
Надо средствами numpy/pandas, без циклов. Производительность имеет значение.
a = np.array([1, 1, 1, 1, 1, -1, -1, -1, -1, -1, -1, 1, 1, 1, 1, 1, 1])
Ожидаемый результат:
[1, 2, 3, 4, 5, 1, 2, 3, 4, 5, 6, 1, 2, 3, 4, 5, 6]
Пример решения с циклом:
res = np.full_like(a, np.nan)
counter = 1
...