From 639686cde9e43b4ad18f9b56960da3f5e8c697dd Mon Sep 17 00:00:00 2001 From: shivesh Date: Sat, 26 May 2018 10:50:03 -0500 Subject: Sat May 26 10:50:03 CDT 2018 --- utils/plot.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'utils/plot.py') 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) -- cgit v1.2.3