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

Question

在标题中，用listl索引具有(n,)布尔ndarrayar的len(l) == ar.shape[0]最有效的方法是什么？你知道吗

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]]