如何创建用于递归生成迭代函数的函数

Question

我目前有一些看起来像这样的 Python 代码：

for set_k in data:
    for tup_j in set_k:
        for tup_l in tup_j:

问题是，我希望嵌套 for 语句的数量根据用户输入而有所不同。如果我想创建一个生成 n 个像上面那样的 for 语句的函数，我该怎么做呢？

Answer 1

def nfor(data, n=1):
    if n == 1:
        yield from iter(data)
    else:
        for element in data:
            yield from nfor(element, n=n-1)

演示：

>>> for i in nfor(['ab', 'c'], n=1):
...     print(i)
...     
ab
c
>>> for i in nfor(['ab', 'c'], n=2):
...     print(i)
...     
a
b
c