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

时间:2017-02-06 09:40:09

标签: python list numpy

与标题一样,使用list l布置(n,) ndarray的{​​{1}} ar索引len(l) == ar.shape[0] npm install npm@latest -g 的最有效方式是什么? }?

1 个答案:

答案 0 :(得分:0)

好的,让我们列出你可以尝试的习语。把它们计算在一起你必须自己做,因为它取决于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]]
相关问题