的Python：高效遍歷多維列表

Question

我使用for循環遍歷二維列表：

def itr(lpic, lH, lW, x, y): 
    '''lpic=2D-Array; lH=Row_count; lW=Column_count;''' 
    stack = [] 
    range_x = range(x-1, x+2) 
    range_y = range(y-1, y+2) 
    append = stack.append 
    for i in range_x: 
       if 0<=i<lH:#i is a valid index *Updated 
        for j in range_y: 
         if (0<=j<lW) and (lpic[i][j]=="0"): 
          lpic[i][j] = "1" 
          append([i, j]) 
    return stack 

我想知道是否有更好的方法做同樣的用的python2.5。

Answer 1

有您的代碼兩個簡單的優化：

1. 使用xrange而不是爲range。這將防止生成兩個臨時列表。

2. 在xrange的參數中使用min和max以省略外循環中的'if'。所以，你的代碼將看起來像：

 
    def itr(lpic, lH, lW, x, y): 
    '''lpic=2D-Array; lH=Row_count; lW=Column_count;''' 
    stack = [] 
    range_x = xrange(max(0,x-1), min(lH,x+2)) 
    range_y = xrange(max(0,y-1), min(lW,y+2)) 
    append = stack.append 
    for i in range_x: 
     for j in range_y: 
      if lpic[i][j]=="0": 
       lpic[i][j] = "1" 
       append([i, j]) 
    return stack 

這將稍微增加性能。

Answer 2

不是。在Python 2.6中，如果你想稍微壓縮代碼，你可以使用itertools.product()來將它變成一個for循環，但是一般效率根本不會改變 - 你仍然會有循環的迭代次數N*M。

import itertools 

def itr(lpic, lH, lW, x, y): 
    '''lpic=2D-Array; lH=Row_count; lW=Column_count;''' 
    stack = [] 
    range_x = range(x-1, x+2) 
    range_y = range(y-1, y+2) 
    append = stack.append 
    for i,j in itertools.product(range_x, range_y): 
     if 0 <= i < lh and 0 <= j < lW and lpic[i][j]=="0": 
      lpic[i][j] = "1" 
      append([i, j]) 
    return stack