segmentation of ROS numpy array

I am new to ROS and PCL. I have a lidar video data which I converted to numpy array of [24000,3]. I want to perform a segmentation on that data based on height. Can someone point me towards resources which can help me to achieve that? I have attached a image that I get from open3D, I want to extract red as a separate array and all other data to other array.