python：递归扩展列表理解

Question

我一直在环顾四周，但我还没有完全解决我的特定问题。 使我的问题具体化的是，我仅尝试使用列表理解。

题：

使用以下代码：

[print(k,l,m) for k in range(3) for l in range(2) for m in range(1)]

我得到的输出：

0 0 0
0 1 0
1 0 0
1 1 0
2 0 0
2 1 0

这适用于n = 3或任何硬编码的n。 但是，我想做的是：

[print(k1,k2,...,kn) for k1 in range(n) for k2 in range(n-1) .. for kn in range(1)].

我希望那里的人能立即看到答案-谢谢！

编辑解决方案发布在下面，但也要清楚我想要的格式，这是我使用它来打印编码元组的方式：

n=3
[print(x) for x in itertools.product(*[list(range(i)) for i in range(n,0,-1)])]

（为简单起见，没有在范围上使用反向）。

Answer 1

如果允许您使用itertools.product：

n=3
list(itertools.product(*[list(range(i)) for i in reversed(range(1, n+1))]))

Answer 2

最近有人问一个非常相似的问题： “ [如何]在不使用itertools.product情况下确定itertools.product硬币的所有组合？” 我为此提供了答案。 我不会将其视为重复项，因为我相信itertools.product最适合您的特定问题。

但是，您可能想知道itertools.product 如何解决问题。 或者，您可能想在递归扩展中编码一个不同的转换。 下面，我将分享使用Python生成器的一种可能的解决方案

def product (*iters):
  def loop (prod, first = [], *rest):
    if not rest:
      for x in first:
        yield prod + (x,)
    else:
      for x in first:
        yield from loop (prod + (x,), *rest)
  yield from loop ((), *iters)

for prod in product ("ab", "xyz"):
  print (prod)

# ('a', 'x')
# ('a', 'y')
# ('a', 'z')
# ('b', 'x')
# ('b', 'y')
# ('b', 'z')

因为product接受可迭代列表，所以任何可迭代输入都可以在产品中使用。 它们甚至可以按此处所示进行混合

print (list (product (['@', '%'], range (2), "xy")))
# [ ('@', 0, 'x')
# , ('@', 0, 'y')
# , ('@', 1, 'x')
# , ('@', 1, 'y')
# , ('%', 0, 'x')
# , ('%', 0, 'y')
# , ('%', 1, 'x')
# , ('%', 1, 'y')
# ]

我们可以制作一个程序foo ，以提供您问题中发布的输出

def foo (n):
  def build_ranges (m):
    if m == 0:
      return []
    else:
      return [ range (m) ] + build_ranges (m - 1)
  yield from product (*build_ranges (n))


for prod in foo (3):
  print (prod)

# (0, 0, 0)
# (0, 1, 0)
# (1, 0, 0)
# (1, 1, 0)
# (2, 0, 0)
# (2, 1, 0)

或使用解构分配为产品的各个元素创建绑定

for (x,y,z) in foo (3):
  print ("x", x, "y", y, "z", z)

# x 0 y 0 z 0
# x 0 y 1 z 0
# x 1 y 0 z 0
# x 1 y 1 z 0
# x 2 y 0 z 0
# x 2 y 1 z 0

我对所链接问题的回答中讨论了该product实施的其他质量。 如果您有兴趣了解如何在不使用生成器的情况下执行此操作，那么我也将提供该问题的纯功能解决方案。