Mercurial > repos > mvdbeek > add_input_name_as_column
diff add_input_name_as_column.py @ 0:bfebefdd5ba4
Initial Commit.
author | mvdbeek |
---|---|
date | Mon, 29 Dec 2014 21:26:29 +0100 |
parents | |
children | 07d2cbf43b51 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/add_input_name_as_column.py Mon Dec 29 21:26:29 2014 +0100 @@ -0,0 +1,25 @@ +import sys +import argparse + +def Parser(): + the_parser = argparse.ArgumentParser(description="add label to last column of file") + the_parser.add_argument('--input', required=True, action="store", type=str, help="input tabular file") + the_parser.add_argument('--output', required=True, action="store", type=str, help="output file path") + the_parser.add_argument('--label', required=True, action="store", type=str, help="label to add in last column") + the_parser.add_argument('--header', action="store", type=str, help="column label for last column") + args = the_parser.parse_args() + return args + +args=Parser() + +input=open(args.input) +output=open(args.output, 'w') +for i,line in enumerate(input): + line=line.strip('\n') + if (i==0) and (args.header!=None): + line=line+'\t'+args.header + else: + line=line+'\t'+args.label + print >>output, line +input.close() +output.close()