python - 深入——yield from inside yield

Question

是否有一个单行代码来获取生成器并生成该生成器中的所有元素？例如:

def Yearly(year):
  yield YEARLY_HEADER
  for month in range(1, 13):
    # How can I rewrite the following two lines?
    for line in Monthly(month):
      yield line
  yield YEARLY_FOOTER

def Monthly(month):
  yield MONTHLY_HEADER
  for day in range(31):  # Yes, this is wrong
    yield 'Day %d' % day
  yield MONTHLY_FOOTER

也许有更好的方法来重写整个方法？

Answer 1

import itertools

def Yearly(year):
  return itertools.chain(*[(YEARLY_HEADER,)] + 
                         [Monthly(m) for m in range(1, 13)] + 
                         [(YEARLY_FOOTER,)])

基本上，将 YEARLY_HEADER 和 YEARLY_FOOTER 制作成迭代器，它们可以与每月迭代器链接起来。