make animation deal with nan

scripts/plot_cci_buoy_error.py

@@ -125,39 +125,43 @@ for hour in hours_list_to_plot:
     cci_to_plot = daily_sst_zonal_mean.isel(time=len(sst.time)-1)
     plot_sst.plot_sst_map(cci_to_plot, ax=ax, vmin=-7, vmax=7)
 
-    for date in dates_list:
-        try:
-            buoy_data = df_buoy[date.strftime('%Y-%m-%d')][0]
-            xx = buoy_data.lon.values
-            yy = buoy_data.lat.values
-            s = 30 / buoy_data.age.values
-            temp_diff = buoy_data['cci_sst'] - buoy_data['cci_sst_1x1']
-            temp = temp_diff.values
-
-            cmap = plt.get_cmap('PiYG_r', 20)
-
-            img = ax.scatter(xx, yy, s=s,
-                             c=temp, cmap=cmap, vmin=-1, vmax=1, edgecolors='k',
-                             linewidths=0.5,
-                             transform=crs.PlateCarree())
-        except IndexError:
-            pass
-
-    # Now adding the colour-bar
-    cax, kw = colorbar.make_axes(ax, location='top', pad=0.07, aspect=30)
-    out = f.colorbar(img, cax=cax, **kw)
-    out.set_label('Temperature difference ($^{\circ}$C) \n'
-                  'CCI-SST at buoy coords - CCI-SST mean in a 1x1 gridbox centred at '
-                  'buoy coords',
-                  size='small',
-                  # \n ''cci-sst at buoy coords - cci-sst avg in a 2x2 gridbox \n '
-                  weight='bold')
-
-    f.suptitle(plot_end_date)
-    #ax.set_title()
-
-    plot_name = str(int(year)) + '_' + f"{str(int(month)):0>2}" + f"{str(int(day_to_plot)):0>2}" + f"{str(int(hour)):0>2}"
-    plt.savefig(os.path.join(config['plots_path'], plot_name + '.png'),
-                    bbox_inches='tight', pad_inches=0.25)
-    plt.clf()
-    plt.close(f)
+    if df_buoy is None:
+        plt.close(f)
+        pass
+    else:
+        for date in dates_list:
+            try:
+                buoy_data = df_buoy[date.strftime('%Y-%m-%d')][0]
+                xx = buoy_data.lon.values
+                yy = buoy_data.lat.values
+                s = 30 / buoy_data.age.values
+                temp_diff = buoy_data['cci_sst'] - buoy_data['cci_sst_1x1']
+                temp = temp_diff.values
+
+                cmap = plt.get_cmap('PiYG_r', 20)
+
+                img = ax.scatter(xx, yy, s=s,
+                                 c=temp, cmap=cmap, vmin=-1, vmax=1, edgecolors='k',
+                                 linewidths=0.5,
+                                 transform=crs.PlateCarree())
+            except IndexError:
+                pass
+
+        # Now adding the colour-bar
+        cax, kw = colorbar.make_axes(ax, location='top', pad=0.07, aspect=30)
+        out = f.colorbar(img, cax=cax, **kw)
+        out.set_label('Temperature difference ($^{\circ}$C) \n'
+                      'CCI-SST at buoy coords - CCI-SST mean in a 1x1 gridbox centred at '
+                      'buoy coords',
+                      size='small',
+                      # \n ''cci-sst at buoy coords - cci-sst avg in a 2x2 gridbox \n '
+                      weight='bold')
+
+        f.suptitle(plot_end_date)
+        #ax.set_title()
+
+        plot_name = str(int(year)) + '_' + f"{str(int(month)):0>2}" + f"{str(int(day_to_plot)):0>2}" + f"{str(int(hour)):0>2}"
\ No newline at end of file
+        plt.savefig(os.path.join(config['plots_path'], plot_name + '.png'),
+                        bbox_inches='tight', pad_inches=0.25)
+        plt.clf()
+        plt.close(f)
\ No newline at end of file