from peers.utils import sanetext
parser = ArgumentParser(description=__doc__)
parser.add_argument('datafn', metavar='data')
parser.add_argument('simufn', metavar='simulation')
parser.add_argument('-output', dest='outfn')
parser.add_argument('-batch', action='store_true')
if __name__ == '__main__':
ns = parser.parse_args()
if ns.batch:
import matplotlib
matplotlib.use('PDF')
import matplotlib.pyplot as pp
from peers.graphics import kdeplot
data = np.load(ns.datafn)
simu = np.load(ns.simufn)
dataset = os.path.basename(os.path.splitext(ns.datafn)[0])
kdeplot(simu, num=100, c='r')
pp.hist(data, bins=50, fc='none', ec='k', normed=True)
pp.xlabel(r'$\tau$ (log-days)')
pp.ylabel(r'density')
pp.title(sanetext(dataset))
if ns.outfn:
pp.savefig(ns.outfn, format='pdf')
print 'output written to %s' % ns.outfn
pp.show()
from matplotlib.backends.backend_pdf import PdfPages
import matplotlib.pyplot as pp
from peers.graphics import kdeplot
parser = ArgumentParser(description=__doc__)
parser.add_argument('datafile', metavar='data')
parser.add_argument('outfile', metavar='output')
parser.add_argument('-every', type=int)
if __name__ == '__main__':
ns = parser.parse_args()
sim = np.load(ns.datafile)
pages = PdfPages(ns.outfile)
files = sorted(sim.files, key=key)
sl = slice(None,None,ns.every)
npages = len(files[sl])
print 'plotting %d pages from %s to %s' % (npages, ns.datafile, ns.outfile)
try:
for i,f in enumerate(files[sl]):
kdeplot(sim[f], c='r')
pp.xlabel(r'$\tau$ (log-days)')
pp.ylabel(r'density')
pp.title(f)
pages.savefig()
print 'page %d of %d: %s' % (i+1, npages, f)
finally:
pages.close()
