diff --git a/extreme_data/meteo_france_data/scm_models_data/abstract_study.py b/extreme_data/meteo_france_data/scm_models_data/abstract_study.py
index bf0824c3800b091241182ebd12ed56c366f9d200..5f6e0ba850c4bd2b6958e3995b02c49728d52fda 100644
--- a/extreme_data/meteo_france_data/scm_models_data/abstract_study.py
+++ b/extreme_data/meteo_france_data/scm_models_data/abstract_study.py
@@ -86,7 +86,7 @@ class AbstractStudy(object):
year_to_first_index_and_last_index = OrderedDict()
first_day, last_day = first_day_and_last_day(self.season)
for year, all_days in self.year_to_all_days.items():
- first_index = all_days.index('{}-{}'.format(year-1, first_day))
+ first_index = all_days.index('{}-{}'.format(year - 1, first_day))
last_index = all_days.index('{}-{}'.format(year, last_day))
year_to_first_index_and_last_index[year] = (first_index, last_index)
return year_to_first_index_and_last_index
@@ -95,7 +95,7 @@ class AbstractStudy(object):
def year_to_days(self) -> OrderedDict:
year_to_days = OrderedDict()
for year, (start_index, last_index) in self.year_to_first_index_and_last_index.items():
- year_to_days[year] = self.year_to_all_days[year][start_index:last_index+1]
+ year_to_days[year] = self.year_to_all_days[year][start_index:last_index + 1]
return year_to_days
@cached_property
@@ -226,6 +226,24 @@ class AbstractStudy(object):
year_to_annual_maxima[year] = time_serie.argmax(axis=0)
return year_to_annual_maxima
+ @cached_property
+ def massif_name_to_annual_maxima_ordered_index(self):
+ massif_name_to_annual_maxima_ordered_index = OrderedDict()
+ for i, (massif_name, years) in enumerate(self.massif_name_to_annual_maxima_ordered_years.items()):
+ ordered_index = [self.year_to_annual_maxima_index[year][i] for year in years]
+ massif_name_to_annual_maxima_ordered_index[massif_name] = ordered_index
+ return massif_name_to_annual_maxima_ordered_index
+
+ @cached_property
+ def massif_name_to_annual_maxima_ordered_years(self):
+ massif_name_to_annual_maxima_ordered_years = OrderedDict()
+ for i, massif_name in enumerate(self.study_massif_names):
+ maxima = np.array([self.year_to_annual_maxima[year][i] for year in self.ordered_years])
+ annual_maxima_ordered_index = np.argsort(maxima)
+ annual_maxima_ordered_years = [self.ordered_years[idx] for idx in annual_maxima_ordered_index]
+ massif_name_to_annual_maxima_ordered_years[massif_name] = annual_maxima_ordered_years
+ return massif_name_to_annual_maxima_ordered_years
+
""" Annual total """
@property
@@ -269,7 +287,7 @@ class AbstractStudy(object):
# 1: to the start_index and last_index of the season
# 2: to the massifs for the altitude of interest
first_index, last_index = self.year_to_first_index_and_last_index[year]
- daily_time_serie = daily_time_serie[first_index:last_index+1, self.column_mask]
+ daily_time_serie = daily_time_serie[first_index:last_index + 1, self.column_mask]
assert daily_time_serie.shape == (len(self.year_to_days[year]), len(self.study_massif_names))
year_to_daily_time_serie_array[year] = daily_time_serie
return year_to_daily_time_serie_array
diff --git a/extreme_data/meteo_france_data/scm_models_data/crocus/crocus.py b/extreme_data/meteo_france_data/scm_models_data/crocus/crocus.py
index 554fda4ef409d0feb412e648b1dc41001a7a7103..2aed6b08c410043666aba2004947e340a10b4e1e 100644
--- a/extreme_data/meteo_france_data/scm_models_data/crocus/crocus.py
+++ b/extreme_data/meteo_france_data/scm_models_data/crocus/crocus.py
@@ -61,10 +61,12 @@ class CrocusSnowLoadTotal(Crocus):
def __init__(self, *args, **kwargs):
Crocus.__init__(self, TotalSnowLoadVariable, *args, **kwargs)
+
class CrocusSnowLoad1Day(CrocusSweTotal):
def __init__(self, *args, **kwargs):
Crocus.__init__(self, RecentSnowLoadVariableOneDay, *args, **kwargs)
+
class CrocusSnowLoad3Days(CrocusSweTotal):
def __init__(self, *args, **kwargs):
Crocus.__init__(self, RecentSnowLoadVariableThreeDays, *args, **kwargs)
diff --git a/extreme_data/meteo_france_data/scm_models_data/crocus/crocus_variables.py b/extreme_data/meteo_france_data/scm_models_data/crocus/crocus_variables.py
index 8b0eac94fdb6d2a0c4c2feb25de1b422af092708..ad044cad418ce8b73d1acdedd2ff351814c58f92 100644
--- a/extreme_data/meteo_france_data/scm_models_data/crocus/crocus_variables.py
+++ b/extreme_data/meteo_france_data/scm_models_data/crocus/crocus_variables.py
@@ -60,9 +60,11 @@ class AbstractSnowLoadVariable(CrocusVariable):
snow_pressure = self.snow_load_multiplication_factor * super().daily_time_serie_array
return snow_pressure
+
class RecentSnowLoadVariableOneDay(AbstractSnowLoadVariable, CrocusRecentSweVariableOneDay):
NAME = 'Snow load last 1 day'
+
class RecentSnowLoadVariableThreeDays(AbstractSnowLoadVariable, CrocusRecentSweVariableThreeDays):
NAME = 'Snow load last 3 days'
diff --git a/projects/contrasting_trends_in_snow_loads/short_snow_load_partition.py b/projects/contrasting_trends_in_snow_loads/short_snow_load_partition.py
new file mode 100644
index 0000000000000000000000000000000000000000..5405c2948c96dbb77faffeaafcffb3f9f3a280f7
--- /dev/null
+++ b/projects/contrasting_trends_in_snow_loads/short_snow_load_partition.py
@@ -0,0 +1,95 @@
+from collections import OrderedDict
+
+import matplotlib as mpl
+import pandas as pd
+import numpy as np
+
+from extreme_data.meteo_france_data.scm_models_data.crocus.crocus_variables import AbstractSnowLoadVariable
+from extreme_data.meteo_france_data.scm_models_data.visualization.main_study_visualizer import \
+ ALL_ALTITUDES_WITH_20_STATIONS_AT_LEAST
+
+mpl.use('Agg')
+mpl.rcParams['text.usetex'] = True
+mpl.rcParams['text.latex.preamble'] = [r'\usepackage{amsmath}']
+
+from extreme_data.meteo_france_data.scm_models_data.safran.safran import SafranSnowfall5Days, SafranSnowfall7Days, \
+ SafranRainfall1Day, SafranRainfall3Days, \
+ SafranRainfall5Days, SafranRainfall7Days, SafranRainfall, SafranSnowfall
+
+from extreme_data.meteo_france_data.scm_models_data.crocus.crocus import CrocusSnowLoad3Days, \
+ CrocusSnowLoad5Days, CrocusSnowLoad7Days, CrocusSnowLoad1Day, Crocus
+from extreme_data.meteo_france_data.scm_models_data.safran.safran import SafranSnowfall1Day, SafranSnowfall3Days
+
+
+def compute_ratio(rainfall, snowfall):
+ total_precipitation = rainfall + snowfall
+ has_some_year_without_precipitation = (total_precipitation == 0).any()
+ if has_some_year_without_precipitation:
+ print('Nan values because we have some year without precipitation')
+ ratio = snowfall / total_precipitation
+ return ratio
+
+
+def df_snow_load_maxima_parittion(rainfall_study: SafranRainfall, snowfall_study: SafranSnowfall,
+ snow_load_study: Crocus):
+ ratios = []
+ for year in snow_load_study.ordered_years:
+ maxima_index = snow_load_study.year_to_annual_maxima_index[year]
+ snowfall = np.diagonal(snowfall_study.year_to_daily_time_serie_array[year][maxima_index])
+ rainfall = np.diagonal(rainfall_study.year_to_daily_time_serie_array[year][maxima_index])
+ ratio = compute_ratio(rainfall, snowfall)
+ ratios.append(pd.Series(ratio))
+ df = pd.concat(ratios, axis=1)
+ df.index = rainfall_study.study_massif_names
+ df.columns = rainfall_study.ordered_years
+ return df
+
+
+def df_snow_load_top_maxima_partition(nb_top_values, rainfall_study: SafranRainfall, snowfall_study: SafranSnowfall,
+ snow_load_study: Crocus):
+ ratios = []
+ for i, (massif_name, ordered_index) in enumerate(
+ snow_load_study.massif_name_to_annual_maxima_ordered_index.items()):
+ ordered_years = snow_load_study.massif_name_to_annual_maxima_ordered_years[massif_name]
+ top_ordered_index = ordered_index[-nb_top_values:]
+ top_ordered_years = ordered_years[-nb_top_values:]
+ # Top values only
+ snowfall = [snowfall_study.year_to_daily_time_serie_array[year][idx, i] for idx, year in
+ zip(top_ordered_index, top_ordered_years)]
+ rainfall = [rainfall_study.year_to_daily_time_serie_array[year][idx, i] for idx, year in
+ zip(top_ordered_index, top_ordered_years)]
+ ratio = compute_ratio(np.array(rainfall), np.array(snowfall))
+ ratios.append(pd.Series(ratio))
+ df = pd.concat(ratios, axis=1).transpose()
+ df.index = rainfall_study.study_massif_names
+ return df
+
+
+def main_snow_load_maxima_partition(year_min, year_max):
+ rainfall_classes = [SafranRainfall1Day, SafranRainfall3Days, SafranRainfall5Days, SafranRainfall7Days]
+ snowfall_classes = [SafranSnowfall1Day, SafranSnowfall3Days, SafranSnowfall5Days, SafranSnowfall7Days]
+ snow_load_classes = [CrocusSnowLoad1Day, CrocusSnowLoad3Days, CrocusSnowLoad5Days, CrocusSnowLoad7Days]
+ classes = list(zip(rainfall_classes, snowfall_classes, snow_load_classes))[:1]
+ nb_top = 5
+ for study_classes in classes:
+ altitude_to_s = OrderedDict()
+ for altitude in ALL_ALTITUDES_WITH_20_STATIONS_AT_LEAST[:]:
+ studies = [study_class(altitude=altitude, year_min=year_min, year_max=year_max) for study_class in
+ study_classes]
+ df = df_snow_load_top_maxima_partition(nb_top, *studies)
+ df2 = df.transpose().describe().transpose()
+ # print(df, '\n')
+ # print(df2.iloc[:, 1:-2])
+ df2.drop(columns=['count', 'std'], inplace=True)
+ s = df2.mean(axis=0)
+ # print(s)
+ altitude_to_s[altitude] = s
+ df_final = pd.DataFrame(altitude_to_s).transpose().round(2)
+ print(nb_top)
+ print(df_final)
+
+
+if __name__ == '__main__':
+ main_snow_load_maxima_partition(1959, 2019)
+ # main_snow_load_maxima_partition(1959, 1989)
+ # main_snow_load_maxima_partition(1990, 2019)