import random

file=open("letter-recognition.data", "r")
xkfile = open('XK-recognition.data', 'w')

for line in file:
    if line[0]=='K' or line[0]=='X':
        line1=line[2:-1]
        if (line[0]=='K'):
            label="-1"
        else:
            label="1"
        line1=line1+','+label+'\n'
        xkfile.write(line1)
file.close()
xkfile.close()

xkfile = open('XK-recognition.data', 'r')
T=[]
for line in xkfile:
    L=line[:-1].split(',')
    T.append(L)

xkfile.close()
random.shuffle(T)

xktrain= open('XK-train.data', 'w')
for i in range(len(T)//3):
    xktrain.write(','.join(T[i])+'\n')
xktrain.close()

xktest=open('XK-test.data', 'w')
for i in range(len(T)//3,len(T)):
    xktest.write(','.join(T[i])+'\n')