python - 从满足 bool 条件的 NumPy 数组中获取(列，行)索引

Question

我正在使用 2D NumPy 数组。我想从满足 bool 条件的二维数组中获取(列、行)索引或(x、y)坐标，如果您更愿意这样想的话。

我可以通过一个简单的例子来解释我正在尝试做的事情的最好方法:

>>> a = np.arange(9).reshape(3, 3)
>>> b = a > 4
>>> b
>>> array([[False, False, False],
           [False, False,  True],
           [ True,  True,  True]], dtype=bool)

此时我有了一个 bool 数组，指示 a > 4 的位置。

此时我的目标是获取值为 True 的 bool 数组的索引。例如，索引 (1, 2)、(2, 0)、(2, 1) 和 (2 , 2) 的值都为 True。

我的最终目标是得到一个索引列表:

>>> indexes = [(1, 2), (2, 0), (2, 1), (2, 2)]

再次强调，上面的代码只是一个简单的例子，但我尝试做的应用程序可能有任意索引，其中 a > 4 而不是基于 arange 和 reshape。

Answer 1

使用numpy.where与 numpy.column_stack :

>>> np.column_stack(np.where(b))
array([[1, 2],
       [2, 0],
       [2, 1],
       [2, 2]])