在了解迭代器类方法和生成器之后,我使用每个习语测试了简单的Fizz Buzz解决方案的性能特征:
>>> from timeit import timeit
>>> timeit('tuple(fizzbuzz.FizzBuzzIterator(10))', 'import fizzbuzz')
13.281935930252075
>>> timeit('tuple(fizzbuzz.fizz_buzz_generator(10))', 'import fizzbuzz')
7.619534015655518
根据timeit
,生成器函数比迭代器类的快<1>快。
我的问题:为什么这个Fizz Buzz生成器明显快于这个Fizz Buzz Iterator类?
class FizzBuzzIterator:
def __init__(self, low, high=None):
if high is None:
self.high = low
self.current = 1
else:
self.high = high
self.current = max(low, 1)
def __iter__(self):
return self
def next(self):
if self.current > self.high:
raise StopIteration
else:
c = self.current
self.current += 1
if (c % 5 + c % 3) == 0:
return 'FizzBuzz'
elif c % 5 == 0:
return 'Buzz'
elif c % 3 == 0:
return 'Fizz'
else:
return str(c)
def fizz_buzz_generator(low, high=None):
if high is None:
high = low
cur = 1
else:
cur = max(low, 1)
while cur <= high:
c = cur
cur += 1
if (c % 5 + c % 3) == 0:
yield 'FizzBuzz'
elif c % 5 == 0:
yield 'Buzz'
elif c % 3 == 0:
yield 'Fizz'
else:
yield str(c)
答案 0 :(得分:6)
因为很明显,生成器比迭代器更有效。
您的第一个解决方案具有以下有趣特征:
第二个解决方案不会执行这些操作,而是yield
s,这意味着语言运行时执行繁重的工作。由于运行时通常以C语言实现,因此可以比第一个解决方案的高级代码更优化。
接下来,您应该考虑您正在进行基准测试的。要获得良好的基准,您应该选择不同的输入尺寸 n ,并观察两种解决方案在不同尺度上的比较。