的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