在与模式匹配的numpy数组中查找所有条目的索引的最简洁方法是什么？

Question

有没有办法巧妙地提取数组中与模式匹配的所有值的索引？

以下示例生成一个元组列表，其索引与数组 A 中为真的元素匹配

import numpy as np

A = np.array([[False, False, False, True],
              [False, True, False, False],
              [False, False, False, False]])
indices = list()
for i, K in enumerate(A):
    for j, L in enumerate(K):
        if L:
            indices.append((i, j))

这行得通，但它看起来很丑陋且“unpythonic”。该解决方案的一个特定缺陷是它依赖于 A 的维度。有更好的方法吗？索引是生成为元组列表还是其他易于访问的东西都没有关系。

Answer 1

你可以使用np.argwhere:

In [1]: A = np.array([[False, False, False, True],
   ...:               [False, True, False, False],
   ...:               [False, False, False, False]])

In [2]: np.argwhere(A == True)
Out[2]: 
array([[0, 3],
       [1, 1]])