scripts/10indiv_format.py at master · tvkent/scripts · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101

#formatting for beagle from ames data

#12feb2014
#tyler kent

def double(inputt, output):

    #specify files to open

    infile=open(inputt, "r")
    linelist=infile.readlines()
    outfile=open(output, "w")

    a=0
    for line in linelist:
        #make list of each word in line
        list=str(linelist[a]).split()
        #add marker & rs# to new line
        newline=list[0]+"\t"+list[1]+"\t"

        x=2
        for i in range(len(list)-2):
            #for each entry from 12 till the end, add 2 copies to newline
            newline=newline+"\t"+list[x]+"\t"+list[x]
            x+=1
        #write each new line to the output file
        outfile.write((newline)+"\n")
        a+=1

    infile.close()
    outfile.close()

def ten(inter):

    import random

    infile=open(inter, "r")
    linelist=infile.readlines()
    outfile=open(inter, "w")

    list=str(linelist[0]).split() #create reference list for column numbers
    used=[]
    numbers=[]
    #run loop size 10
    for i in range(10):
        number=random.randint(2,len(list))
        if number not in used:
                numbers.append(number) #only use unique random column numbers
        while number in used:
                number=random.randint(2,len(list))
                if number not in used:
                    numbers.append(number)
        used.append(number)

    a=0
    for line in linelist:
        #make list of each word in line
        list=str(linelist[a]).split()
        #add marker & rs# to new line
        newline=list[0]+"\t"+list[1]+"\t"
        for number in numbers: #run loop of random numbers from above for each line
            if number==numbers[-1]:
                newline=newline+"\t"+list[number]+"\n" #for last number in line, add newline char
            else:
                newline=newline+"\t"+list[number]+"\t" #tab delimit
        outfile.write(newline)
        a+=1

    infile.close()
    outfile.close()

def markers(inputt, inter):

     #specify files to open
    infile=open(inputt, "r")
    linelist=infile.readlines()

    outfile=open(inter,"w")

    #add 'I' to first line
    line="I\t"+str(linelist[0])
    outfile.write(line)

    #add 'M' to remaining lines
    x=1
    for i in range(len(linelist)-1):
        line="M\t"+str(linelist[x])
        outfile.write(line)
        x+=1

    infile.close()
    outfile.close()


infile=raw_input("File for input (include full path): ")
inter=raw_input("File for intermediate (include full path): ")
outfile=raw_input("File for output (include full path): ")
markers(infile, inter)
ten(inter)
double(inter, outfile)