迭代Python中的多维数组

Question

我在Python中创建了一个多维数组,如下所示:

self.cells = np.empty((r,c),dtype=np.object)

现在我想迭代我的二维数组的所有元素,我不关心顺序.我该如何实现这一目标？

Answer 1

很明显你正在使用numpy.有了numpy,你可以这样做:

for cell in self.cells.flat:
    do_somethin(cell)

Answer 2

如果您需要更改单个单元格的值,则ndenumerate(numpy)是您的朋友.即使你不这样做,它可能仍然是!

for index,value in ndenumerate( self.cells ):
    do_something( value )
    self.cells[index] = new_value

Answer 3

只需迭代一个维度,然后迭代另一个维度.

for row in self.cells:
    for cell in row:
        do_something(cell)

当然,只有两个维度,您可以使用列表推导或生成器表达式将其压缩为单个循环,但这不是非常可伸缩或可读:

for cell in (cell for row in self.cells for cell in row):
    do_something(cell)

如果您需要将其缩放到多个维度并且确实需要平面列表,则可以编写一个flatten函数.

Answer 4

您可以使用enumerate命令获取每个元素的索引以及元素本身:

for (i,row) in enumerate(cells):
  for (j,value) in enumerate(row):
    print i,j,value

i,j包含元素的行和列索引,并且value是元素本身.

Answer 5

这个怎么样:

import itertools
for cell in itertools.chain(*self.cells):
    cell.drawCell(surface, posx, posy)

Answer 6

没有人有一个答案可以在没有 numpy 的情况下从任意多个维度工作，所以我会在这里放一个我使用过的递归解决方案

def iterThrough(lists):
  if not hasattr(lists[0], '__iter__'):
    for val in lists:
      yield val
  else:
    for l in lists:
      for val in iterThrough(l):
        yield val

for val in iterThrough(
  [[[111,112,113],[121,122,123],[131,132,133]],
   [[211,212,213],[221,222,223],[231,232,233]],
   [[311,312,313],[321,322,323],[331,332,333]]]):
  print(val)
  # 111
  # 112
  # 113
  # 121
  # ..

这没有很好的错误检查，但它对我有用