Filtering in numpy array

Question

Want to filter numpy array based on a condition with using only numpy

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

Op:

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

Expected OP:

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

Pointers on my mistake would be great

https://stackoverflow.com/questions/58422690/filtering-a-numpy-array-what-is-the-best-approach — Christophe Roussy, May 05 '21 at 08:50

score 0 · Accepted Answer · answered May 05 '21 at 09:13

0

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

(array([0, 1]),)

which corresponds to the indexes of the true elements.

Your expected output is just given by:

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

answered May 05 '21 at 09:13

dzang

1 Answers1