#!/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)