替代Python中的循环嵌套

Question

我已经读过Python的一个关键信念是flat>嵌套.但是,如果我有几个变量计数,多个for循环的替代方法是什么？我的代码用于计算网格总和,如下所示:

def horizontal():
    for x in range(20):
        for y in range(17):
            temp = grid[x][y: y + 4]
            sum = 0
            for n in temp:
                sum += int(n)
            print sum # EDIT: the return instead of print was a mistype

在我看来,它似乎太嵌套了.首先,考虑到Python中的许多嵌套循环(我之前已经看过2个嵌套循环).其次,如果嵌套太多,编写此代码的另一种方法是什么？

Answer 1

from itertools import product

def horizontal():
    for x, y in product(range(20), range(17)):
        print 1 + sum(int(n) for n in grid[x][y: y + 4])

你应该使用这个sum功能.当然你不能用变量遮住它,所以我把它改成了my_sum