diff options
Diffstat (limited to 'data_analysis/destmaps.py')
| -rw-r--r-- | data_analysis/destmaps.py | 20 |
1 files changed, 10 insertions, 10 deletions
diff --git a/data_analysis/destmaps.py b/data_analysis/destmaps.py index 0725ca7..6d08096 100644 --- a/data_analysis/destmaps.py +++ b/data_analysis/destmaps.py @@ -4,7 +4,7 @@ import cPickle import scipy.misc print "Loading data..." -with open("train_normal.pkl") as f: normal = cPickle.load(f) +with open("train.pkl") as f: normal = cPickle.load(f) print "Extracting x and y" # xes = [c[0] for l in normal for c in l[-1]] @@ -12,21 +12,21 @@ print "Extracting x and y" xes = [l[-1][-1][0] for l in normal if len(l[-1]) > 0] yes = [l[-1][-1][1] for l in normal if len(l[-1]) > 0] -xrg = [-8.75, -8.55] -yrg = [41.05, 41.25] +xrg = [-8.80, -8.50] +yrg = [41.00, 41.30] -print "Doing 1d x histogram" -plt.clf(); plt.hist(xes, bins=1000, range=xrg); plt.savefig("xhist_dest.pdf") -print "Doing 1d y histogram" -plt.clf(); plt.hist(yes, bins=1000, range=yrg); plt.savefig("yhist_dest.pdf") +#print "Doing 1d x histogram" +#plt.clf(); plt.hist(xes, bins=2000, range=xrg); plt.savefig("xhist_dest.pdf") +#print "Doing 1d y histogram" +#plt.clf(); plt.hist(yes, bins=2000, range=yrg); plt.savefig("yhist_dest.pdf") print "Doing 2d histogram" -hist, xx, yy = numpy.histogram2d(xes, yes, bins=2000, range=[xrg, yrg]) +hist, xx, yy = numpy.histogram2d(xes, yes, bins=4000, range=[xrg, yrg]) # import ipdb; ipdb.set_trace() print "Imshow" -plt.clf(); plt.imshow(numpy.log(hist)); plt.savefig("xyhmap_dest.png", dpi=600) +plt.clf(); plt.imshow(numpy.log(hist)); plt.savefig("xyhmap_dest_x.png", dpi=600) print "Imsave" -scipy.misc.imsave("xymap_dest_2.png", numpy.log(hist + 1)) +scipy.misc.imsave("xymap_dest_2_x.png", numpy.log(hist + 1)) |