annotate column_arrange.py @ 0:b84c81b7d762 draft

Uploaded
author bgruening
date Mon, 02 Mar 2015 09:59:32 -0500
parents
children b97f0d9a31ff
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
b84c81b7d762 Uploaded
bgruening
parents:
diff changeset
1 #!/usr/bin/env python
b84c81b7d762 Uploaded
bgruening
parents:
diff changeset
2 import pandas as pd
b84c81b7d762 Uploaded
bgruening
parents:
diff changeset
3 import argparse
b84c81b7d762 Uploaded
bgruening
parents:
diff changeset
4
b84c81b7d762 Uploaded
bgruening
parents:
diff changeset
5
b84c81b7d762 Uploaded
bgruening
parents:
diff changeset
6 parser = argparse.ArgumentParser()
b84c81b7d762 Uploaded
bgruening
parents:
diff changeset
7 parser.add_argument('-i', '--input', help='Tabular Input File Name')
b84c81b7d762 Uploaded
bgruening
parents:
diff changeset
8 parser.add_argument('-o','--output', help='Tabular Output File')
b84c81b7d762 Uploaded
bgruening
parents:
diff changeset
9 parser.add_argument('-c', '--columns',nargs='+', help='Column Headers to Sort By')
b84c81b7d762 Uploaded
bgruening
parents:
diff changeset
10 args=parser.parse_args()
b84c81b7d762 Uploaded
bgruening
parents:
diff changeset
11
b84c81b7d762 Uploaded
bgruening
parents:
diff changeset
12 cols=args.columns
b84c81b7d762 Uploaded
bgruening
parents:
diff changeset
13 table=pd.read_csv(args.input,sep='\t')
b84c81b7d762 Uploaded
bgruening
parents:
diff changeset
14 blist = list(table.columns)
b84c81b7d762 Uploaded
bgruening
parents:
diff changeset
15 for token in cols:
b84c81b7d762 Uploaded
bgruening
parents:
diff changeset
16 blist.remove(token)
b84c81b7d762 Uploaded
bgruening
parents:
diff changeset
17 sorted_table = table[args.columns + blist]
b84c81b7d762 Uploaded
bgruening
parents:
diff changeset
18 # write without index, seperated by tabs
b84c81b7d762 Uploaded
bgruening
parents:
diff changeset
19 sorted_table.to_csv(args.output,sep='\t',index=False)
b84c81b7d762 Uploaded
bgruening
parents:
diff changeset
20