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函数，覆盖它是一个坏主意。

从数组中断并在python

2 个答案: