Partition Dataset by Feature Threshold - Coding problem | OneNoughtOne