Python 类似 Zip 的函数，它迭代列表中的多个项目并返回可能性

发布于07月13日

在以下代码中:

a = [["2022"], ["2023"]]
b = [["blue", "red"], ["green", "yellow"]]
c = [["1", "2", "3"], ["4", "5", "6", "7"], ["8", "9", "10", "11"], ["12", "13"]]

我想要一个输出这个的函数，但对于任何数量的变量:

[
    ["2022", "blue", "1"],
    ["2022", "blue", "2"],
    ["2022", "blue", "3"],
    ["2022", "red", "4"],
    ["2022", "red", "5"],
    ["2022", "red", "6"],
    ["2022", "red", "7"],
    ["2023", "green", "8"],
    ["2023", "green", "9"],
    ["2023", "green", "10"],
    ["2023", "green", "11"],
    ["2023", "yellow", "12"],
    ["2023", "yellow", "13"],
]

我已经用itertools或zip搜索了一个函数来实现这一点，但还没有找到任何结果.

为了澄清，我的用例是迭代嵌套/多级下拉菜单的值(第一个下拉菜单返回选项，每个选项返回不同的下拉菜单，等等).

推荐答案

首先，将第一个参数加入一个列表，每个列表只有一个元素.

然后，对于下一个参数中的每个sublist及其索引i，您 Select 上一个迭代res[i]的第i个列表，并将其添加到aux len(子列表)列表中，其中每个列表都是res[i]，其中一个项目来自sublist.

from itertools import chain

def f(*args):
    res = list(chain.from_iterable([[item] for item in l] for l in args[0]))
    for arg in args[1:]:
        aux = []
        for i, sublist in enumerate(arg):
            aux += [res[i] + [opt] for opt in sublist]
        res = aux
    return res

此外，如果要验证传递给函数的参数是否正确，可以使用以下选项:

def check(*args):
    size = sum(len(l) for l in args[0])
    for arg in args[1:]:
        if len(arg) != size:
            return False
        size = sum(len(l) for l in arg)
    return True