python - 使用python的for语句时如何跳转一些步骤

Question

昨天有个面试，在写python代码实现一些算法的时候，有一个问题。使用 C 可以实现一些逻辑:

void get_some_num(int a[], int length) {       
   int i;
   for(i = 0; i < length - 1; ++i) {
       if(something) i++; // jump a num in the array
       a[i] = 1;
   }
    return some_num;
 }

C语言在for循环中使用i += n语句迭代数组时可以跳转一些元素(比如n)，但是我发现优雅地使用python的for语句很难实现.

我该怎么做？

Answer 1

如果您想以类似于 C 的方式执行此操作，则只需使用 while 循环(任何 for 循环实际上只是毕竟 while 循环的特化):

i = 0
end = 10

while i < end:
    # NOTE: do something with i here

    if i == 5:
        i += 3
    i += 1

或者您可以显式创建迭代器并向前移动(我发现它的可读性差很多):

it = iter(range(0, 10))

for i in it:
    if i == 5:
        for j in range(0, 3):
            i = next(it)
    print(i)