Merge pull request #34 from E3SM-Project/hsianghelee/boolean_array_fix

fix xarray boolean array issue when using where function

chemdyg/templates/chemdyg_TOZ_eq_plot.bash

@@ -125,7 +125,7 @@ lat2 = refer['lat']
 rearth = 6.37122e6
 AREA_in = refer['area'] * rearth * rearth
 
-TOZ_sel = file_in['TCO'].where((lat < -60), drop=True)+file_in['SCO'].where((lat < -60), drop=True)
+TOZ_sel = file_in['TCO'].where((lat.compute() < -60), drop=True)+file_in['SCO'].where((lat.compute() < -60), drop=True)
 AREA_sel = AREA_in.where((lat2 < -60), drop=True)
 AREA_64S = AREA_in.where((lat2 < -64), drop=True).sum()
 AREA_64S = np.array(AREA_64S)
@@ -160,7 +160,7 @@ for i in range(startindex,endindex):
 fig = plt.figure(figsize=(10,5))
 plt.plot(time_range[startindex:endindex],O3_64S)
 plt.title('Total column ozone conc. with equivalent latitude (64S)')
-plt.xlabel('Time')
+plt.xlabel('Time (start in year 2000)')
 plt.ylabel('O3 conc. (DU)',fontsize='large')
 pylab.savefig(pathout+'TOZ_PDF_timeseries.png', dpi=300)
 

chemdyg/templates/chemdyg_temperature_eq_plot.bash

@@ -120,9 +120,9 @@ lat2 = refer['lat']
 rearth = 6.37122e6
 AREA_in = refer['area'] * rearth * rearth
 
-TOZ_h = file_in['T'][:,13].where((lat < -55), drop=True)
-TOZ_m = file_in['T'][:,21].where((lat < -55), drop=True)
-TOZ_l = file_in['T'][:,29].where((lat < -55), drop=True)
+TOZ_h = file_in['T'][:,13].where((lat.compute() < -55), drop=True)
+TOZ_m = file_in['T'][:,21].where((lat.compute() < -55), drop=True)
+TOZ_l = file_in['T'][:,29].where((lat.compute() < -55), drop=True)
 AREA_sel = AREA_in.where((lat2 < -55), drop=True)
 TOZ_h_sel = TOZ_h.sel(time=TOZ_h.time.dt.month.isin([6,7,8,9,10,11,12]))
 TOZ_m_sel = TOZ_m.sel(time=TOZ_m.time.dt.month.isin([6,7,8,9,10,11,12]))