如何创建用于递归生成迭代函数的函数 [英] How to create a function for recursively generating iterating functions

问题描述

我目前有一些类似以下的Python代码:

I currently have a bit of Python code that looks like this:

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

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

The problem is, I'd like the number of nested for statements to differ based on user input. If I wanted to create a function which generated n number of for statements like those above, how might I go about doing that?

推荐答案

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

这篇关于如何创建用于递归生成迭代函数的函数的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！