REDDprocessor/3.sortinterval.py at master · JacksonIsaac/REDDprocessor · 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
import csv
import numpy as np
import os

for root, dirs, files in os.walk(os.curdir):
    for name in files:
        if name.endswith(".dat"):
            #print root
            current = os.path.join(root, name)
            print "Sorting ", current

            with open(current, 'rb') as f, open('redd-'+current[2:]+'.txt', 'wb') as out:
                writer = csv.writer(out)
                count = 1
                # Remove the first line with total number of examples.
                for row in csv.reader(f):
                    try:
                        n_row = [count, "60"]
                        e_type = row
                        s_time = f.next().split(',')
                        dur = f.next().split(',')
                        # print e_type
                        #s_time = [x.replace(",,+",",") for x in s_time]
                        # s_time = [x.replace("^?,,+",",") for x in s_time]
                        # dur = [x.replace("^?,,+",",") for x in dur]
                        # s_time = [x.replace("\n","") for x in s_time]
                        # dur = [x.replace("\n","") for x in dur]
                        # print row
                        # print s_time
                        # print dur
                        l1 = [int(i) for i in e_type]
                        l2 = [int(i) for i in s_time]
                        l3 = [float(i) for i in dur]

                        index = range(len(l2))
                        index.sort(key = l2.__getitem__)

                        l1[:] = [e_type[i] for i in index]
                        l2[:] = [s_time[i] for i in index]
                        l3[:] = [dur[i] for i in index]
                        # print l1
                        # print l2
                        # print l3
                        l2 = [x.replace("\r\n","") for x in l2]
                        l3 = [x.replace("\r\n","") for x in l3]
                        l2 = [x.replace("\n","") for x in l2]
                        l3 = [x.replace("\n","") for x in l3]
                        # print l1
                        # print l2
                        # print l3
                        writer.writerow(n_row)
                        writer.writerow(l1)
                        writer.writerow(l2)
                        writer.writerow(l3)
                        count += 1
                    except(StopIteration):
                        break