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