有没有一种捷径可以在 Python 中连续调用一个函数两次或更多次?例如:
do()
do()
do()
也许像:
3*do()
有没有一种捷径可以在 Python 中连续调用一个函数两次或更多次?例如:
do()
do()
do()
也许像:
3*do()
我会:
for _ in range(3):
do()
这_
是您不关心其值的变量的约定。
您可能还会看到一些人写道:
[do() for _ in range(3)]
但是,这会稍微贵一些,因为它会创建一个包含每次调用的返回值的列表do()
(即使它是None
),然后丢弃结果列表。除非您使用返回值列表,否则我不建议使用它。
您可以定义一个将传递的函数重复 N 次的函数。
def repeat_fun(times, f):
for i in range(times): f()
如果你想让它更加灵活,你甚至可以将参数传递给被重复的函数:
def repeat_fun(times, f, *args):
for i in range(times): f(*args)
用法:
>>> def do():
... print 'Doing'
...
>>> def say(s):
... print s
...
>>> repeat_fun(3, do)
Doing
Doing
Doing
>>> repeat_fun(4, say, 'Hello!')
Hello!
Hello!
Hello!
Hello!
另外三种方法:
(I) 我认为 usingmap
也可能是一种选择,尽管在某些情况下需要生成一个带有 s 的附加列表,None
并且总是需要一个参数列表:
def do():
print 'hello world'
l=map(lambda x: do(), range(10))
(II)itertools
包含可用于迭代其他函数的函数以及https://docs.python.org/2/library/itertools.html
(III) 到目前为止,我认为没有提到使用函数列表(它实际上是最接近最初讨论的语法):
it=[do]*10
[f() for f in it]
或作为一个班轮:
[f() for f in [do]*10]
一个简单的for循环?
for i in range(3):
do()
或者,如果您对结果感兴趣并想收集它们,那么您可以成为 1 班轮:
vals = [do() for _ in range(3)]
我的两分钱:
from itertools import repeat
list(repeat(f(), x)) # for pure f
[f() for f in repeat(f, x)] # for impure f
这是一种不需要使用for
循环或定义中间函数或 lambda 函数(并且也是单线)的方法。该方法结合了以下两种思想:
使用可选的 sentinel 参数调用iter()
内置函数,以及
使用推进迭代器步骤的itertools
秘诀n
(参见秘诀consume()
)。
把这些放在一起,我们得到:
next(islice(iter(do, object()), 3, 3), None)
(作为哨兵传递的想法object()
来自这个公认的 Stack Overflow 答案。)
这是交互式提示中的样子:
>>> def do():
... print("called")
...
>>> next(itertools.islice(iter(do, object()), 3, 3), None)
called
called
called
from itertools import repeat, starmap
results = list(starmap(do, repeat((), 3)))
请参阅 itertools 模块中的repeatfunc配方,它实际上更强大。如果您只需要调用该方法但不关心返回值,您可以在 for 循环中使用它:
for _ in starmap(do, repeat((), 3)): pass
但这越来越难看。
您可以使用itertools.repeat
withoperator.methodcaller
调用__call__
函数的方法N次。这是执行此操作的生成器函数的示例:
from itertools import repeat
from operator import methodcaller
def call_n_times(function, n):
yield from map(methodcaller('__call__'), repeat(function, n))
使用示例:
import random
from functools import partial
throw_dice = partial(random.randint, 1, 6)
result = call_n_times(throw_dice, 10)
print(list(result))
# [6, 3, 1, 2, 4, 6, 4, 1, 4, 6]
您可以尝试如下所示的while循环;
def do1():
# Do something
def do2(x):
while x > 0:
do1()
x -= 1
do2(5)
因此调用 do1 函数 5 次。