python - 在 Python 中编写多重嵌套 for 循环的最佳方法是什么

Question

我正在编写需要使用嵌套循环的代码，如下所示:

for r in range(m):
   for s in range(r+1, m):
      for t in range(s+1, m):
         for u in range(t+1, m):
            for v in range(u+1, m):
                  arr.append([r,s,t,u,v])

但是这种传统的嵌套循环看起来非常难看。有没有办法用更少的行执行相同的操作？

我查看了 itertools.product，但我无法得到我想要的东西，因为我的循环的所有开始/结束索引都取决于之前的级别。

Answer 1

您可以使用 itertools.combinations，第二个参数是您要执行的循环数。

from itertools import combinations
for item in combinations("ABCD", 3):
    print item

输出

('A', 'B', 'C')
('A', 'B', 'D')
('A', 'C', 'D')
('B', 'C', 'D')

所以，有了列表理解，整个代码就变成了这样

[list(item) for item in combinations("ABCD", 3)]