Дан одномерный массив числовых значений, насчитывающий N элементов. После каждого отрицательного элемента вставить новый элемент, равный квадрату этого отрицательного элемента. Питон

Предыдущий
вопрос Следующий
вопрос

Решение данной задачи на Python:

def insert_squares(arr):
i = 0
while i < len(arr):
if arr[i] < 0:
arr.insert(i+1, arr[i]**2)
i += 2
else:
i += 1
return arr
# Пример использования
arr = [1, -2, 3, -4, 5]
new_arr = insert_squares(arr)
print(new_arr)

При запуске данного кода будет получен следующий результат:

[1, -2, 4, 3, -4, 16, 5]