在给定的numpy矩阵中查找等于零的元素的索引

Question

我正在尝试使用numpy作为3D整形游戏问题的一部分，在3 * 3整数矩阵中查找非零元素的索引。我意识到np.where是这种情况的一个不错的选择，并对其进行了尝试，我得到的输出看起来仍然不正确。你能帮我编码这部分吗？我在下面给出了我的部分代码。

输入：s，尺寸为3 * 3的整数矩阵 例如：

输出：m，可能的下一个动作的列表，其中每个下一个动作是一个（r，c）元组，其中r表示行号，c表示列号。

示例：

[代码]

m = np.where(s==0)

Answer 1

这是一个快速的解决方案：

import numpy as np

s = np.matrix('0, 0, 0; 0, 1, 0; 0, 0, 0')

m = np.where(s==0)
m = list(zip(m[0], m[1]))

print(m)

s是输入矩阵，您可以在其中看到中间的正方形，然后像您一样使用np.where()，产生两个数组，然后使用zip()将它们组合成元组，然后list()将输出转换成有效动作的元组列表。