python - 简化 Python 迭代

Question

每次我尝试解决一些数学问题，例如找到特定数量因子的特定乘积时，我都会在 Python 中执行此操作

for x in xrange(1,10):
    for y in xrange(1,10):
        for z in xrange(1,10):
           product = x * y * z
           if product == 36:
               print "factors : {0},{1},{2}".format(x,y,z)

在这个例子中，它非常简单，可以快速完成工作，但我想知道你们是否知道一种更简单或更简单的方法来编写它。关于如何在不使用那么多迭代或一遍又一遍地重复几乎相同的代码的情况下执行此操作的任何想法。这些显然是针对 3 个因素，但我添加的因素越多，代码就越长、重复得越多。关于如何简化这种简单类型问题的代码的任何想法？ 谢谢

Answer 1

Itertool 的 cartesian product模拟多个嵌套for循环的效果。

import itertools

for x, y, z in itertools.product(range(1,10), range(1,10), range(1,10)):
    product = x * y * z
    if product == 36:
        print "factors : {0},{1},{2}".format(x,y,z)

结果:

factors : 1,4,9
factors : 1,6,6
factors : 1,9,4
(...etc)

如果每个 x、y 和 z 的范围始终相同，您可以只指定一次:

for x, y, z in itertools.product(range(1,10), repeat=3):

如果您厌倦了在 product = 行中输入无数星号，您可以使用 reduce 将任意数量的参数相乘:

for factors in itertools.product(range(1,3), repeat=10):
    product = reduce(lambda x, y: x*y, factors)

一旦您的格式字符串变得笨拙，您可以依靠 join 将因素串在一起:

if product == 512:
    #use `map` to turn the factors into strings, first
    print "factors: " + ",".join(map(str, factors))