valid = intersect1d(valid, where(U_mag > 0)[0]) valid = intersect1d(valid, where(U_ob_v > 1e-9)[0]) valid = intersect1d(valid, where(abs(Mb_v) < 200)[0]) valid = intersect1d(valid, where(S_v > -100)[0]) valid = intersect1d(valid, where(Ts_v > 100)[0]) valid = intersect1d(valid, where(h_v > 0)[0]) valid = intersect1d(valid, where(S_v - B_v > 60)[0]) valid = intersect1d(valid, where(adot_v > -100)[0]) valid_f = Function(Q) valid_f_v = valid_f.vector().array() valid_f_v[valid] = 1.0 valid_f.vector().set_local(valid_f_v) valid_f.vector().apply('insert') rignot = DataFactory.get_gre_rignot() drg = DataInput(rignot, gen_space=False) betaMax = 200.0 ini_f = Function(Q) ini_f.vector()[:] = ini plotIce(drg, ini_f, name='ini', direc='images/stats/' + file_n, cmap='gist_yarg', scale='log', numLvls=12, tp=False, tpAlpha=0.5, show=False, umin=1.0, umax=betaMax) Ubar_avg_f = Function(Q) Ubar_avg_f.vector()[:] = Ubar_avg plotIce(drg, Ubar_avg_f, name='Ubar_avg', direc='images/stats/' + file_n, title=r'$\Vert \mathbf{\bar{u}}_{\bar{bv}} \Vert$', cmap='gist_yarg', scale='log', numLvls=12, tp=False,