76 lines
1.7 KiB
Python
76 lines
1.7 KiB
Python
"""
|
|
迭代器 - __iter__ / __next__
|
|
itertools - 生成可迭代序列的工具模块
|
|
"""
|
|
import itertools
|
|
|
|
from math import sqrt
|
|
|
|
|
|
def is_prime(num):
|
|
"""判断素数"""
|
|
for factor in range(2, int(sqrt(num)) + 1):
|
|
if num % factor == 0:
|
|
return False
|
|
return True
|
|
|
|
|
|
class PrimeIter(object):
|
|
"""素数迭代器"""
|
|
|
|
def __init__(self, min_value, max_value):
|
|
assert 2 <= min_value <= max_value
|
|
self.min_value = min_value - 1
|
|
self.max_value = max_value
|
|
|
|
def __iter__(self):
|
|
return self
|
|
|
|
def __next__(self):
|
|
self.min_value += 1
|
|
while self.min_value <= self.max_value:
|
|
if is_prime(self.min_value):
|
|
return self.min_value
|
|
self.min_value += 1
|
|
raise StopIteration()
|
|
|
|
|
|
class FibIter(object):
|
|
"""斐波那契数迭代器"""
|
|
|
|
def __init__(self, num):
|
|
self.num = num
|
|
self.a, self.b = 0, 1
|
|
self.idx = 0
|
|
|
|
def __iter__(self):
|
|
return self
|
|
|
|
def __next__(self):
|
|
if self.idx < self.num:
|
|
self.a, self.b = self.b, self.a + self.b
|
|
self.idx += 1
|
|
return self.a
|
|
raise StopIteration()
|
|
|
|
|
|
def main():
|
|
# for val in itertools.permutations('ABCD'):
|
|
# print(val)
|
|
# for val in itertools.combinations('ABCDE', 3):
|
|
# print(val)
|
|
# for val in itertools.product('黑红梅方', range(1, 14)):
|
|
# print(val)
|
|
# fib_iter = FibIter(20)
|
|
# print('===>', next(fib_iter))
|
|
# print('===>', next(fib_iter))
|
|
# for val in fib_iter:
|
|
# print(val)
|
|
prime_iter = PrimeIter(2, 100000)
|
|
for val in prime_iter:
|
|
print(val)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|