在 numpy 数组中过滤

Question

想要根据仅使用 numpy

的条件过滤 numpy array

sample = ["aple","mangp", "orange"]
np.where("p" in sample)

Op:

(array([], dtype=int64),)

Expected OP:

(array([1,1,0], dtype=int64),)

如果能指出我的错误就太好了

Answer 1

sample = ["aple", "mangp", "orange"]
np.where(["p" in s for s in sample])

(array([0, 1]),)

对应真实元素的索引

您的预期输出由以下公式给出：

np.array(["p" in s for s in sample]).astype(int)