迭代时如何获得二维 Numpy 数组中的位置？

Question

我有一个 2D Numpy 数组，它表示我在 Python3.7 中制作的基于图块的游戏中的关卡地图。我的数组是 50 行 x 50 列，每个值都是一个整数，代表一种精灵类型（各种墙壁部件、地板等）。当我遍历数组并找到一个墙精灵（例如）时，我想知道该值的行号和列号，以便可以直接引用它。

这似乎是一件很简单的事情，但我似乎找不到合适的 Numpy 函数来遍历 2D 数组并返回它找到的每个数组的 位置[row, col]。然后我可以使用数组位置计算它在屏幕上的相对位置，然后测试与.colliderect()的碰撞。

编辑：为了更清楚起见，我添加了一个事实，即我需要将数组中的每个整数值与墙壁部分（也是整数）的List 进行比较，如果它在列表中，则返回数组位置.

TL;DR如何遍历 Numpy 2darray 并为每个匹配返回 [row, col] 而不是值？

Answer 1

只需使用np.argwhere。

例子：

>>> import numpy as np
>>> a = np.zeros((50, 50))
>>> a[3, 4] = 1
>>> a[4, 5] = 1
>>> for i, j in np.argwhere(a == 1): print(i, j)
... 
(3, 4)
(4, 5)

（上面的示例是使用 python 2 运行的，因此输出中显示的元组，但在其他方面与 python 3 完全相同。）

Answer 2

我认为您可以使用嵌套的 for 循环和如下数据结构：

假设 map 是一个二维 numpy 数组，其中 1 代表墙壁类型，2 代表地板类型。

代码：

locationsToSprites = dict()#Data structure that stores the locations of the sprite type. 
for row in range(0, len(map)): #looping through all rows
    for col in range(0, len(map[row])):#looping through all columns of a row
        if(map[row][col] == 1):
            locationsToSprites[(row, col)] = 'the wall type'
        elif(map[row][col] == 2):
            locationsToSprites[(row, col)] = 'the floor type'

现在，由于使用了字典来存储精灵的位置，我们只需打印字典的键即可获取所有位置。