get prime numbers from numpy array

Viewed 22878

I have a numpy array like,

nums = np.array([17, 18, 19, 20, 21, 22, 23])

How do I filter out the prime numbers from this array in a pythonic manner? I know to do a simple filtering like,

nums[nums > 20]   #array([21, 22, 23])

Is there a way to pass a lambda function for filtering?

Expected output: array([17, 19, 23])

5 Answers
Related