tools/plot-values/scatter.py
author Dmitriy Morozov <dmitriy@mrzv.org>
Mon Jul 25 23:21:29 2011 -0700 (6 months ago)
branchdev
changeset 244 66235db8d8b7
parent 13896030f8d2f2c
permissions -rwxr-xr-x
Added cohomology/candidates counter to i/t/cohomology-persistence.hpp
     1 #!/usr/bin/env python
     2 
     3 from    pylab           import scatter, show, cm, colorbar, axes, savefig
     4 from    itertools       import izip
     5 from    sys             import argv, exit
     6 import  os.path         as     osp
     7 
     8 
     9 def plot(val1_fn, val2_fn, outfn = None):
    10     values1 = []
    11     with open(val1_fn) as fp:
    12         for line in fp.xreadlines():
    13             values1.append(float(line.split()[1]))
    14     
    15     values2 = []
    16     with open(val2_fn) as fp:
    17         for line in fp.xreadlines():
    18             values2.append(float(line.split()[1]))
    19     
    20     values1 = [v % 1. for v in values1]
    21     values2 = [v % 1. for v in values2]
    22     print min(values1), max(values2), min(values1), min(values2)
    23 
    24     scatter(values1, values2, s=10)
    25     axes().set_aspect('equal')
    26     if not outfn:
    27         show()
    28     else:
    29         savefig(outfn)
    30 
    31 if __name__ == '__main__':
    32     if len(argv) < 3:
    33         print "Usage: %s VALUES1 VALUES2 [OUTPUT]" % argv[0]
    34         exit()
    35 
    36     val1_fn = argv[1]
    37     val2_fn  = argv[2]
    38     
    39     outfn = None
    40     if len(argv) > 3: 
    41         outfn = argv[3]
    42 
    43     plot(val1_fn, val2_fn, outfn)