在python中的2D列表（矩阵）范围内进行迭代

Question

我想遍历2D列表。 我了解可以使用以下方法做到这一点：

for j in range(columns):
     for i in range(rows):
        print depth[i][j],

但是有没有更美丽的方法。 我试过了：

for (x in range(rows)) and (y in range(columns)):
     print(depth[x][y])

但这给了我一个IndentationError

谢谢

编辑：如果我想为x和y设置边界怎么办。 假设我有一个4x4矩阵（使用列表表示）。 对于给定的元素，我想在其周围获得3x3矩阵，但我希望保持在原始矩阵的范围内。 假设我在（3,1），取x = range(3-2, 3+3) = [1,2,3,4,5] y = range(1-2,1+3) = [-1, 0, 1, 2, 3] x = range(3-2, 3+3) = [1,2,3,4,5]和y = range(1-2,1+3) = [-1, 0, 1, 2, 3]将我带到原始矩阵的范围之外。

Answer 1

您可以使用itertools.product ，也可以编写一个双循环generator-expression，例如：

for (i,j) in ( (i,j) for i in range(rows) for j in range(columns) ):
    print depth[i][j]

要么

for d in ( depth[i][j] for i in range(rows) for j in range(columns) ): ...

另外，如果您实际上不需要索引，而仅需要遍历depth的值 ，则可以执行以下操作：

for row in depth:
   for v in row:
      print

即使您确实需要索引，使用enumerate还是更Python化：

for row_index, row in enumerate(depth):
   for col_index, v in enumerate(row):
      print 'depth[%d][%d]=%s' % (row_index, col_index, v)

Answer 2

这可以通过以下方式完成：

for depth in (depth[x][y] for x in range(rows) for y in range(columns)):
    print depth

编辑：

如果您想要界线，则可以使用min和max内置函数：

for x in range( max(x_min,col-2), min(x_max,col+3) ):
    #do stuff