Random Splitting Of Data Python