如何用列表推导式实现可变层数的递减范围循环输出？

嘿，这个问题我之前也琢磨过——想用列表推导式实现动态层数的嵌套循环，确实有点 tricky，毕竟原生的列表推导式语法是静态的，层数写死在代码里没法随n变化。不过我们可以借助Python标准库的工具，实现和你想要的效果一致的写法，同时保留列表推导式的风格。

核心思路 #

你要的本质是生成依次递减的range的笛卡尔积，然后打印每个积的元素。直接写可变层数的列表推导式做不到，但我们可以用itertools.product来动态生成这些组合，再用列表推导式遍历打印。

解决方案代码 #

import itertools

n = 4  # 这里可以替换成任意正整数
[print(*combination) for combination in itertools.product(*(range(k) for k in range(n, 0, -1)))]

代码解释 #

• range(n, 0, -1)会生成从n到1的递减序列（比如n=3时是3,2,1），我们用它来动态创建每个循环的range对象：(range(k) for k in range(n, 0, -1))是一个生成器，输出range(3), range(2), range(1)（n=3时）。
• itertools.product(*...)会把这些range作为参数，生成所有可能的笛卡尔积组合——这就相当于你手动写n层嵌套循环的效果。
• 最后用列表推导式遍历每个组合，print(*combination)把组合里的元素拆成独立参数传入print，和你硬编码的print(k,l,m)效果完全一致。

额外优化 #

如果不想生成一个全是None的无用列表（因为print函数返回None），可以改成生成器表达式并消耗掉，更节省内存：

import itertools

n = 3
any(print(*comb) for comb in itertools.product(*(range(k) for k in range(n, 0, -1))))

内容的提问来源于stack exchange，提问作者Richard