How to get indexes of items greater or less than each item in a NumPy array without using a loop?

Question

I am writing a Decision Tree algorithm from scratch, right now I\'m trying to split the data into groups where each group contains values that are greater or equal or less than