Как запустить определенную часть функции в коде, когда код был выполнен x раз в python?

Question

У меня есть код, который имеет несколько функций, и число итераций кода равно 10.

def vectfit_auto(f, s, n_poles=5, n_iter=10,loss_ratio=1e-2, rcond=-1,):

for _ in range(n_iter):
    poles, Zeros, H = vectfit_step(f, s, poles)


    poles_list.append(poles)

Я хочу добавить несколько строк в vectfit_step (одну измои функции), как показано ниже для модификации:

from iteration number of 5 to 10
do something

Я хочу, чтобы код выполнялся как прежде, и моя модификация будет применяться только от номера итерации 5 до конца.Как я могу это сделать?Спасибо

Answer 1

#i takes values between begin and (end - 1)
for i in range(begin, end):
   do_something()

#In your case start = 5 and end = 11
for i in range(5, 10+1):
   do_something(i)

#You might use _, if you are not interested in the value of i
for _ in range(5, 11):
   do_something()

Answer 2

Как прокомментировано, вы можете включить оператор if в свой цикл и запускать его только после того, как вы запустите основной цикл определенное количество раз.

for i in range(6): # 11 - 5
    if i == 5:
        for i in range(5):
            do_something()
     # main code here