清理动态生成的嵌套元组

Question

我有一个类，其__iter__() 方法返回动态生成数据的itertools.product()。 这些数据是任意嵌套字典的笛卡尔积，现在我需要将它展平，但是以一种流线型的方式，聚合中间迭代器。

我正在尝试以下任何一种：

1. 修改__iter__()以处理内部元组：

    class Explosion:
        ...
        def __iter__(self):
            return product(*self.fragments)
   

2. 将其封装在另一个对象中以处理转换，但这不太理想：

    class CleanOutput:
        def __init__(self, it):
            self.it = it
   
        def next(self):
            for x in self.it:
                yield ?
   
    class Explosion:
        ...
        def __iter__(self):
            return CleanOutput(product(*self.fragments))
   

好吧，算法确实有效，但问题是最后的拆包，例如：

(11, ..., (10.7, 104.75, ('N', True, False, 'B2B'), 99.01, ...), 1, 'SP', 7).

看看所有的嵌套！如何在real-time 中删除它？在生成时... 我正在寻找一种检索方式：

(11, ..., 10.7, 104.75, 'N', True, False, 'B2B', 99.01, ..., 1, 'SP', 7).

最好和最快的方法是什么？ 谢谢！

编辑

实际上，我真正想要的是列表推导式或生成器表达式，甚至是另一个生成器，因为我需要将它包含在一个可调用对象中，截取itertools.product() 本身的输出.我不只是需要一种方式来清理这些元组。所以它不是重复的。

Answer 1

def gen(data):
   for item in data:
      if isinstance(item, tuple):
          for nested in gen(item):
              yield nested
      else:
          yield item

未经测试，但应该可以工作。

Answer 2

这并不容易，必须使用递归，但要与主要的__iter__ 方法分开。我就是这样结束的。 现在还有一个递归生成器_merge，由另一个生成器_flatten调用：

class Explosion:
    # ...

    def __iter__(self):
        def _flatten(container):
            def _merge(t):
                for te in t:
                    if isinstance(te, tuple):
                        for ite in _merge(te):
                            yield ite
                    else:
                        yield te

            for t in container:
                yield tuple(_merge(t))

        return _flatten(product(*self.fragments))

查看_flatten() 函数的使用示例：

>>> list(itertools.product([1,2],[3,(4,(5,6))]))
[(1, 3), (1, (4, (5, 6))), (2, 3), (2, (4, (5, 6)))]
>>> list(_flatten(itertools.product([1,2],[3,(4,(5,6))])))
[(1, 3), (1, 4, 5, 6), (2, 3), (2, 4, 5, 6)]