python - 生成元组以在一个 for 循环中迭代

Question

我想通过一次生成所有迭代器来将三个 for 循环减少为一个。基本上我有一本使用元组作为键的字典。我知道我能做到

for aa in range(limitA):
    for bb in range(limitB):
        for cc in range(limitC):
            do_stuff(my_dict[(aa,bb,cc)])

但是有没有一种方法可以减少代码行数呢？有点像

for aa,bb,cc in range(limitA), range(limitB), range(limitC):
    do_stuff(my_dict[(aa,bb,cc)])

这将返回错误“ValueError:需要超过 2 个值才能解包”

如果一般情况下无法做到这一点，limitA == limitB == limitC 的特殊情况是否有解决方案？

Answer 1

使用itertools.product() :

from itertools import product

for aa, bb, cc in product(range(limitA), range(limitB), range(limitC)):

其中 limitA == limitB == limitC 这可以简化为:

for aa, bb, cc in product(range(limitA), repeat=3):