2021-02-15 00:58:46 +01:00
|
|
|
# tag::ARITPROG_ITERTOOLS[]
|
|
|
|
|
import itertools
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def aritprog_gen(begin, step, end=None):
|
|
|
|
|
first = type(begin + step)(begin)
|
|
|
|
|
ap_gen = itertools.count(first, step)
|
2021-09-16 03:48:08 +02:00
|
|
|
if end is None:
|
|
|
|
|
return ap_gen
|
|
|
|
|
return itertools.takewhile(lambda n: n < end, ap_gen)
|
2021-02-15 00:58:46 +01:00
|
|
|
# end::ARITPROG_ITERTOOLS[]
|
