diff options
| author | shivesh <s.p.mandalia@qmul.ac.uk> | 2018-05-26 10:50:03 -0500 |
|---|---|---|
| committer | shivesh <s.p.mandalia@qmul.ac.uk> | 2018-05-26 10:50:03 -0500 |
| commit | 639686cde9e43b4ad18f9b56960da3f5e8c697dd (patch) | |
| tree | 45c7d9b7e98ea0094aa169604dacd72ae89ed867 /utils/plot.py | |
| parent | f6645a19974f87166460a9baceade6817212a319 (diff) | |
| download | GolemFlavor-639686cde9e43b4ad18f9b56960da3f5e8c697dd.tar.gz GolemFlavor-639686cde9e43b4ad18f9b56960da3f5e8c697dd.zip | |
Sat May 26 10:50:03 CDT 2018
Diffstat (limited to 'utils/plot.py')
| -rw-r--r-- | utils/plot.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/utils/plot.py b/utils/plot.py index 4e1a58b..7a82f2b 100644 --- a/utils/plot.py +++ b/utils/plot.py @@ -14,6 +14,7 @@ import socket from copy import deepcopy import numpy as np +import numpy.ma as ma from scipy import interpolate import matplotlib as mpl @@ -269,7 +270,7 @@ def plot_statistic(data, outfile, outformat, args, scale_param, label=None): print 'data', data print 'data.shape', data.shape - scales, statistic = data.T + scales, statistic = ma.compress_rows(data).T tck, u = interpolate.splprep([scales, statistic], s=0) scales, statistic = interpolate.splev(np.linspace(0, 1, 1000), tck) print 'scales', scales @@ -496,7 +497,7 @@ def plot_sens_fixed_angle_pretty(data, outfile, outformat, args): bc_limit = best_limits[dim] # ax.axvline(x=np.log10(bc_limit[1]), color=bc_limit[2], alpha=0.7, linewidth=1.5) - scales, statistic = data[idim][isrc][ian].T + scales, statistic = ma.compress_rows(data[idim][isrc][ian]).T tck, u = interpolate.splprep([scales, statistic], s=0) scales, statistic = interpolate.splev(np.linspace(0, 1, 1000), tck) min_idx = np.argmin(scales) |