参数数量未知的手动产品

Question

以下示例给出了相同的结果：

一种。

product = []
for a in "abcd":
    for b in "xy":
        product.append((a,b))

B．

from itertools import product
list(product("abcd","xy"))

当我不知道参数n时如何计算例A中的笛卡尔积？

原因我问这个：

考虑这段代码：

allocations = list(product(*strategies.values()))
for alloc in allocations:
    PWC[alloc] = [a for (a,b) in zip(help,alloc) if coalitions[a] >= sum(b)]

strategies词典的值是元组列表， help是辅助变量（每个alloc具有相同长度的列表），而coalitions是另一个在help中分配给元组一些数值的词典。

由于对策略值进行了排序，因此我知道，在某个alloc之后，if语句将不再成立。 由于分配是一个很大的列表，因此，如果可以使用示例算法A，我将避免大量的比较和大量的求和。

Answer 1

你可以做：

items = ["abcd","xy"]

from itertools import product
list(product(*items))

列表items可以包含任意数量的字符串，使用product进行计算将为您提供这些字符串的笛卡尔乘积。

请注意，您不必将其变成列表-您可以对其进行迭代，并在不再希望继续时停止：

for item in product(*items):
    print(item)
    if condition:
        break

Answer 2

如果您只是想在达到特定条件后中止分配，并且想要避免从笛卡尔乘积中为这些元素生成所有元素，那么根本就不用列出所有组合。

itertools.product是惰性的 ，这意味着它一次只会生成一个笛卡尔积的单个值。 因此，您无需生成所有元素，也不需要再比较这些元素。 只是不要在结果上调用list() ，因为那样会迭代整个序列并将所有可能的组合存储在内存中：

allocations = product(*strategies.values())
for alloc in allocations:
    PWC[alloc] = [a for (a,b) in zip(help,alloc) if coalitions[a] >= sum(b)]

    # check whether you can stop looking at more values from the cartesian product
    if someCondition(alloc):
        break

重要的是要注意itertools.product如何生成值，遵循的模式。 它基本上等效于以下内容：

for a in firstIterable:
    for b in secondIterable:
        for c in thirdIterable:
            …
                for n in nthIterable:
                    yield (a, b, c, …, n)

因此，您从可迭代对象的左侧获得越来越多的模式。 因此，请确保以正确指定中断条件的方式订购可迭代对象。