Python函数可以使用生成器并将生成器返回到其生成的输出的子集吗？

Question

假设我有这样一个生成器函数：

import random
def big_gen():
  i = 0
  group = 'a'
  while group != 'd':
    i += 1
    yield (group, i)
    if random.random() < 0.20:
      group = chr(ord(group) + 1)

示例输出可能是： （“a”，1），（“a”，2），（“a”，3），（“a”，4），（“a”，5），（“a”，6），（“a”，7），（“a”，8），（“b”，9），（“c”，10），（“c”，11），（“c”，12），（“c”，13）

我想把这个分成三组：A组，B组，C组。我想为每一组配备一个发电机。然后我把生成器和群字母传递给一个子函数。子功能示例：

^{pr2}$

预期输出为：

These numbers are in group a:
1
2
3
4
5
6
7
8
These numbers are in group b:
9
These numbers are in group c:
10
11
12
13

如何在不更改big_gen（）或printer（）的情况下完成此操作，并避免同时将整个组存储在内存中？（在现实生活中，群体是巨大的

Answer 1

你这里有点小问题。您希望函数to printer（）为每个组获取一个生成器，但实际上您有一个生成所有组的生成器。在我看来，你有两个选择：

1）改变big_gen（）以产生发电机：

import random
def big_gen():
  i = 0
  group = 'a'
  while group != 'd':
    def gen():
        i += 1
        yield i
        if random.random() < 0.20:
            group = chr(ord(group) + 1)
    yield group, gen

 from itertools import imap
 imap(lambda a: printer(*a), big_gen())

2）更改printer（）以保持状态并在组更改时通知（保留原来的big_gen（）函数）：

^{pr2}$