summaryrefslogtreecommitdiffstats
path: root/data/combined/combo.py
blob: bd212748be6411971937bf0aa07450b2e3dbafed (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
#!/usr/bin/python
import os
import sys
import pickle

runs = []
names = pickle.load(open(sys.argv[1]))
labels = ''
dirs = ['24-1-2012-16-56-57','27-1-2012-10-45-29']
for arg in sys.argv[2:]:
    run = open(arg)
    labels = run.readline().strip('#').strip()
    runs += [[run, arg.split('/')[-2]]]

print '#dir,'+labels
for lines, sub_dir in runs:
    for line in lines:
        line = line.strip().split(',')
        line[0] = str(names.index(line[0]) + 1)
        line.insert(0,str(dirs.index(sub_dir) + 1))
        print ','.join(line)