python - 从数组中中断并在 python 中运行另一个函数

Question

我的代码:

def exit():
   return "Exit" 

arr = [1, 2, 3, 4, 5, 6, 7, 8, 9, 0]

def func():
   for element in arr:
      print element

我想在数组的每 3 个元素之后运行 exit()，然后从最后一个元素恢复。

Answer 1

您可以使用 enumerate和模:

def exit():
    return "Exit"


arr = [1, 2, 3, 4, 5, 6, 7, 8, 9, 0]

def func():
    for ind, element in enumerate(arr, 1):
        print element
        if ind % 3 == 0:
            print(exit())

输出:

1
2
3
Exit
4
5
6
Exit
7
8
9
Exit
0

如果你真的想一次使用 n 个元素，可以将它们传递给一个函数:

from itertools import islice

arr = [1, 2, 3, 4, 5, 6, 7, 8, 9, 0]

def func():
    it = iter(arr)
    for sli in iter(lambda: list(islice(it, 3)),[]):
        print(sli) # pass sli to function
        print(exit())

func()

输出:

[1, 2, 3]
Exit
[4, 5, 6]
Exit
[7, 8, 9]
Exit
[0]
Exit