NumPy 数组上项目的坐标

Question

我有一个 NumPy 数组：

[[  0.   1.   2.   3.   4.]
 [  7.   8.   9.  10.   4.]
 [ 14.  15.  16.  17.   4.]
 [  1.  20.  21.  22.  23.]
 [ 27.  28.   1.  20.  29.]]

我想快速找到特定值的坐标并避免数组上的 Python 循环。例如号码4 开启：

row 0 and col 4
row 1 and col 4
row 2 and col 4

并且搜索函数应该返回一个元组：

((0,4),(1,4),(2,4))

这可以直接通过 NunmPy 的函数来完成吗？

Answer 1

如果a 是您的数组，那么您可以使用：

ii = np.nonzero(a == 4)

或

ii = np.where(a == 4)

如果你真的想要一个元组，你可以从数组的元组转换为元组的元组，但是 numpy 函数的返回值便于对数组进行其他操作。

转换为 OP 规范的元组：

tuple(zip(*ii))

Answer 2

a = numpy.array([[  0.,  1.,  2.,  3.,  4.],
                 [  7.,  8.,  9., 10.,  4.],
                 [ 14., 15., 16., 17.,  4.],
                 [  1., 20., 21., 22., 23.],
                 [ 27., 28.,  1., 20., 29.]])
print numpy.argwhere(a == 4.)

打印

[[0 4]
 [1 4]
 [2 4]]

适用于浮点比较的常见注意事项。