在 Python 中使用 numpy bool 数组索引列表的有效方法

Question

如标题所示，索引具有(n,)布尔ndarray ar且len(l) == ar.shape[0]的list l的最有效方法是什么？

Answer 1

好的，让我们列出您可以尝试的习语。 给它们计时，你必须自己做，因为这取决于ar的大小和稀疏性，参见。 @Imanol Luengo 的评论

np.array(l)[ar] # output is an array
[e for e,m in zip(l, ar) if m]
[l[i] for i in np.where(ar)[0]]