如何以pythonic方式查看for循环中的for循环

Question

我从 itertools.product 中找到这段代码来查找列表的唯一组合

args = [["a","b"], ["a", "c", "d"], ["h"]]
pools = [tuple(pool) for pool in args]

for pool in pools:
    result = [x + [y] for x in result for y in pool]

给出：

print(result)
[['a', 'a', 'h'], ['a', 'c', 'h'], ['a', 'd', 'h'], ['b', 'a', 'h'], ['b', 'c', 'h'], ['b', 'd', 'h']]

现在我想知道是否有一种方法可以使用 for 循环以“正常”方式编写它？我设法用 if 语句将它重写为一个 for 循环，如下所示：

[s for s in p if s != 'a']

等于：

s = []
for x in p:
    if x != 1:
        s.append(x)

但是我还没有设法在 for 循环中为 for 循环执行此操作...我对此很陌生，所以我猜必须有某种方法可以做到这一点，但我没有怎么看。有人怎么做吗？

Answer 1

我认为你可以继续这个趋势，例如：

[(x,y) for x in [0,1,2,3,4,5] if x < 3 for y in [0,1,2,3,4,5] if 2 < y if x + y == 4]

相当于（将每个for 和if 放在一个新行上）：

s = []
for x in [0,1,2,3,4,5]:
    if x < 3:
        for y in [0,1,2,3,4,5]:
            if 2 < y:
                if x + y == 4:
                    s.append((x,y))

对于问题中的示例，列表推导式中的result 指的是result 的旧值，因此在创建新的result 时需要保留该值：

result = [[]]
for pool in pools:
    old_result = result # remember the old result
    result = [] # build the new result with this variable
    for x in old_result:
        for y in pool:
            result.append(x + [y])

或者，您可以在不同的变量中构建新的result 并将result 设置为它：

result = [[]]
for pool in pools:
    new_result = [] # build the new result with this variable
    for x in result:
        for y in pool:
            new_result.append(x + [y])
    result = new_result # update our current result

还有另一个例子here。