Python 3.x numpy阵列中的滤波

要根据仅使用numpy的条件筛选

numpy数组

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

Op:

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

Expected OP:

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

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

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

（数组（[0，1]），）

它对应于真实元素的索引

您的预期输出如下所示：

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