我的代码:
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(),然后从最后一个元素恢复。
请您参考如下方法:
您可以使用 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
