Python中嵌套循環的替代方案

我讀過Python的一個關鍵信念，那就是flat> nested。但是，如果我有幾個變量計數，多個for循環的替代方法是什麼？我的代碼是用於計數網格和以及去如下：Python中嵌套循環的替代方案

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個嵌套循環）。其次，如果嵌套過於嚴重，那麼編寫此代碼的另一種方法是什麼？

來源

2012-12-17 davenz

您正在尋找['itertools.product']（http：//docs.pyth on.org/2/library/itertools.html#itertools.product）。 –

你確定你的return語句在循環內部嗎？ – 0xc0de

0xc0de，這是錯誤的（我的意思是打印），謝謝你指出它 – davenz

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

來源

2012-12-17 05:02:48

我相信你在'20'後面錯過了一個正確的表情。 – recursive

@recursive，謝謝修復 –

我會使用'xrange'而不是'range'。我也同意循環內的'return'是一個錯誤。 – Balthamos

grid = [range(20) for i in range(20)] 
sum(sum(1 + sum(grid[x][y: y + 4]) for y in range(17)) for x in range(20))

上述輸出13260，對於代碼的第一行所產生的特定網格。它使用sum()三次。最裏面的總和將grid[x][y: y + 4]中的數字相加，再加上問題中代碼中顯示的稍微奇怪的初始值sum = 1。中間總和爲17個可能的y值加上這些值。外部和將可能的x值的中間值相加。

如果電網的元素是字符串而非數字，更換
sum(grid[x][y: y + 4])
與
sum(int(n) for n in grid[x][y: y + 4]

來源

2012-12-17 05:10:28

您可以使用字典來優化性能顯著

這是另一個例子：

locations = {} 
for i in range(len(airports)): 
    locations[airports["abb"][i][1:-1]] = (airports["height"][i], airports["width"][i]) 

for i in range(len(uniqueData)): 
    h, w = locations[uniqueData["dept_apt"][i]] 
    uniqueData["dept_apt_height"][i] = h 
    uniqueData["dept_apt_width"][i] = w

來源

2016-10-12 09:08:07

Python中嵌套循環的替代方案

回答

相關問題