comparison scatterplot.py @ 2:53eb2c412783 draft

planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/tools/scatterplot commit de7140295cce07e1bc1697e51dab4271c8d7a8a6
author devteam
date Fri, 18 Dec 2015 19:14:12 -0500
parents c30eeaaac86d
children 1dacc8aeb452
comparison
equal deleted inserted replaced
1:4afd68392187 2:53eb2c412783
1 #!/usr/bin/env python 1 #!/usr/bin/env python
2 #Greg Von Kuster 2 # Greg Von Kuster
3 3
4 import sys 4 import sys
5 from rpy import * 5
6 from numpy import array
7 from rpy import r
8
6 9
7 def stop_err(msg): 10 def stop_err(msg):
8 sys.stderr.write(msg) 11 sys.stderr.write(msg)
9 sys.exit() 12 sys.exit()
13
10 14
11 def main(): 15 def main():
12 16
13 in_fname = sys.argv[1] 17 in_fname = sys.argv[1]
14 out_fname = sys.argv[2] 18 out_fname = sys.argv[2]
27 invalid_column = 0 31 invalid_column = 0
28 i = 0 32 i = 0
29 for i, line in enumerate( file( in_fname ) ): 33 for i, line in enumerate( file( in_fname ) ):
30 valid = True 34 valid = True
31 line = line.rstrip( '\r\n' ) 35 line = line.rstrip( '\r\n' )
32 if line and not line.startswith( '#' ): 36 if line and not line.startswith( '#' ):
33 row = [] 37 row = []
34 fields = line.split( "\t" ) 38 fields = line.split( "\t" )
35 for column in columns: 39 for column in columns:
36 try: 40 try:
37 val = fields[column] 41 val = fields[column]
38 if val.lower() == "na": 42 if val.lower() == "na":
39 row.append( float( "nan" ) ) 43 row.append( float( "nan" ) )
40 else: 44 else:
41 row.append( float( fields[column] ) ) 45 row.append( float( fields[column] ) )
42 except: 46 except:
43 valid = False 47 valid = False
52 break 56 break
53 else: 57 else:
54 valid = False 58 valid = False
55 skipped_lines += 1 59 skipped_lines += 1
56 if not first_invalid_line: 60 if not first_invalid_line:
57 first_invalid_line = i+1 61 first_invalid_line = i + 1
58 62
59 if valid: 63 if valid:
60 matrix.append( row ) 64 matrix.append( row )
61 65
62 if skipped_lines < i: 66 if skipped_lines < i:
63 try: 67 try:
64 r.pdf( out_fname, 8, 8 ) 68 r.pdf( out_fname, 8, 8 )
65 r.plot( array( matrix ), type="p", main=title, xlab=xlab, ylab=ylab, col="blue", pch=19 ) 69 r.plot( array( matrix ), type="p", main=title, xlab=xlab, ylab=ylab, col="blue", pch=19 )
66 r.dev_off() 70 r.dev_off()
67 except Exception, exc: 71 except Exception, exc:
68 stop_err( "%s" %str( exc ) ) 72 stop_err( "%s" % str( exc ) )
69 else: 73 else:
70 stop_err( "All values in both columns %s and %s are non-numeric or empty." % ( sys.argv[3], sys.argv[4] ) ) 74 stop_err( "All values in both columns %s and %s are non-numeric or empty." % ( sys.argv[3], sys.argv[4] ) )
71 75
72 print "Scatter plot on columns %s, %s. " % ( sys.argv[3], sys.argv[4] ) 76 print "Scatter plot on columns %s, %s. " % ( sys.argv[3], sys.argv[4] )
73 if skipped_lines > 0: 77 if skipped_lines > 0: