Specifically change all elements in a numpy array based on a condition without iterating over each e