从数组中断并在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和modulo:

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

Answer 2

def func():
   for i, element in enumerate(arr, 1):
      print element
      if not i%3:
          print(exit())

但是，我对代码的某些部分提出了质疑。以下是一些注意事项：arr是一个python列表，而不是一个数组，所以这个名称不是最好的名称。您的exit函数只是返回字符串"Exit"，您确定它执行了它应该执行的操作吗？我正在打印exit()的返回值，因为简单地调用它没有任何用处。此外，最好编写func，使其将列表作为参数。最后，exit已经是一个内置的Python函数，覆盖它是个坏主意。