In [11]:

    

import math
import random

with open('shuttle.trn', 'r') as input_data, \
    open('shuttle.data', 'w') as output_data, \
    open('shuttle_toy.data', 'w') as output_toy_data:
    for line in input_data:
        output_data.write(line)
        if math.floor(10*random.random()) == 1:
            output_toy_data.write(line)


    

In [12]:

    

with open('shuttle.data', 'r') as data, \
    open("shuttle_test.data", "w") as test_file, \
    open("shuttle_train.data", "w") as train_file:
    for line_c in data:
        if math.floor(30*random.random()) == 1:
            test_file.write(line_c)
        else:
            train_file.write(line_c)
            
with open('shuttle_toy.data', 'r') as data, \
    open("shuttle_toy_test.data", "w") as test_file, \
    open("shuttle_toy_train.data", "w") as train_file:
    for line_c in data:
        if math.floor(30*random.random()) == 1:
            test_file.write(line_c)
        else:
            train_file.write(line_c)