Splitting data into Train and Test Set | Data Preprocessing | Machine Learning | Python | Data Magic