Python代码阅读(第29篇):使用初始化种子和迭代函数生成列表

Python 代码阅读合集介绍: 为什么不推荐Python初学者直接看项目源码

本篇阅读的代码实现使用一个初始化种子和迭代函数,通过嵌套函数对初始化种子进行迭代,最终生成一个列表。

本篇阅读的代码片段来自于30-seconds-of-python

unfold

def unfold(fn, seed):
  def fn_generator(val):
    while True: 
      val = fn(val[1])
      if val == False: break
      yield val[0]
  return [i for i in fn_generator([None, seed])]

# EXAMPLES
f = lambda n: False if n > 50 else [-n, n + 10]
unfold(f, 10) # [-10, -20, -30, -40, -50]

unfold函数接收一个迭代函数和一个初始化种子,生成一个列表。迭代函数fn接受一个参数seed,并且必须总是返回一个有两个元素的列表[value,nextSeed],或者返回False来终止生成器函数。unfold函数内部定义生成器fn_generator。该生成器使用while循环调用迭代器函数并产生值,直到它返回False。最后使用列表推导式返回生成器函数产生的列表。

生成器是一个用于创建迭代器的简单而强大的工具。 它的写法类似于标准的函数,但当它们要返回数据时会使用yield语句。 每次在生成器上调用next()时,它会从上次离开的位置恢复执行(它会记住上次执行语句时的所有数据值)。

fn_generator是一个生成器,在其中while语句的最后,使用yield返回当前while循环中val[0]的值。下一次调用的时候,会继续下一个循环,继续返回下一个循环中val[0]的值,直到迭代函数fn返回false结束while循环。

unfold函数的return语句中,使用列表推导式不断调用fn_generator生成器,直到生成器退出循环,形成一个列表并返回。

你可能感兴趣的:(Python代码阅读(第29篇):使用初始化种子和迭代函数生成列表)