Python 循环的列表理解

Question

我使用了很多 N 维数组，编写这样的缩进代码很痛苦，而且我知道有些代码可以用列表理解和内联语句替换。例如：

for x in (0,1,2,3):
    for y in (0,1,2,3):
        if x < y:
            print (x, y, x*y)

可以替换为：

print [(x, y, x * y) for x in (0,1,2,3) for y in (0,1,2,3) if x < y]

但是我如何更改操作而不是打印来执行其他操作，例如：

total = x+y

所以我想做的是：

[(total+=x+y) for x in (0,1,2,3) for y in (0,1,2,3) if x < y]

但是这不起作用 有没有

一种聪明的方法可以做到这一点而不是:

for x in (0,1,2,3):
        for y in (0,1,2,3):
            if x < y:
                total+=x+y

Answer 1

使用 numpy。这使您可以使用像向量一样相加的数组：

x = numpy.arange(3)
y = numpy.arange(3)
total = x + y

通过修改后的问题，还添加对 sum 的调用

total = numpy.sum(x+y)

Answer 2

归约函数直接将集合项减少为单个项。您可以在此处阅读有关它们的更多信息，但这应该对您有用：

total=reduce(lambda x,y:x+y,range(4))

或者

total=reduce(lambda x,y:x+y,(0,1,2,3))

Answer 3

另一种可能性是：

for x,y in ((x,y) for x in (0,1,2,3) for y in (0,1,2,3) if x < y):
  print (x, y, x * y)

通过这种方式，您可以迭代在列表理解中使用的任何内容，而无需实际创建理解列表（如果您明白我的意思；）如果理解列表很大，可能太大以至于饱和甚至不饱和放在内存里，蛮方便的

Answer 4

sum 在这里工作：

total = sum(x+y for x in (0,1,2,3) for y in (0,1,2,3) if x < y)

Answer 5

作为编写 N 层深度循环的替代方法，您可以使用 itertools.product ()：

In [1]: import itertools as it

In [2]: for x, y in it.product((0,1,2,3),(0,1,2,3)):
   ...:     if x < y:
   ...:         print x, y, x*y

0 1 0
0 2 0
0 3 0
1 2 2
1 3 3
2 3 6

这自然延伸到 N 维。