Commit 7bc067a3 authored by Le Roux Erwan's avatar Le Roux Erwan
Browse files

[PAPER 1] add label to trend plot

parent d6287e58
No related merge requests found
Showing with 17 additions and 8 deletions
+17 -8
import matplotlib as mpl import matplotlib as mpl
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from experiment.meteo_france_data.scm_models_data.visualization.study_visualization.main_study_visualizer import \
ALL_ALTITUDES_WITHOUT_NAN
from experiment.paper_past_snow_loads.paper_utils import paper_study_classes, paper_altitudes, \ from experiment.paper_past_snow_loads.paper_utils import paper_study_classes, paper_altitudes, \
load_altitude_to_visualizer load_altitude_to_visualizer
from experiment.paper_past_snow_loads.result_trends_and_return_levels.eurocode_visualizer import \ from experiment.paper_past_snow_loads.result_trends_and_return_levels.eurocode_visualizer import \
...@@ -17,9 +19,10 @@ mpl.rcParams['text.latex.preamble'] = [r'\usepackage{amsmath}'] ...@@ -17,9 +19,10 @@ mpl.rcParams['text.latex.preamble'] = [r'\usepackage{amsmath}']
def minor_result(altitude): def minor_result(altitude):
"""Plot trends for a single altitude to be fast""" """Plot trends for a single altitude to be fast"""
visualizer = StudyVisualizerForNonStationaryTrends(CrocusSnowLoadTotal(altitude=altitude), multiprocessing=True) visualizer = StudyVisualizerForNonStationaryTrends(CrocusSnowLoadTotal(altitude=altitude), multiprocessing=True,
)
visualizer.plot_trends() visualizer.plot_trends()
plt.show() # plt.show()
def intermediate_result(altitudes, massif_names=None, def intermediate_result(altitudes, massif_names=None,
...@@ -59,8 +62,9 @@ def major_result(): ...@@ -59,8 +62,9 @@ def major_result():
if __name__ == '__main__': if __name__ == '__main__':
# major_result() # major_result()
# intermediate_result(paper_altitudes) # intermediate_result(ALL_ALTITUDES_WITHOUT_NAN)
minor_result(altitude=600) intermediate_result(paper_altitudes)
# minor_result(altitude=600)
# intermediate_result(altitudes=[1500, 1800], massif_names=['Chartreuse'], # intermediate_result(altitudes=[1500, 1800], massif_names=['Chartreuse'],
# uncertainty_methods=[ConfidenceIntervalMethodFromExtremes.ci_mle, # uncertainty_methods=[ConfidenceIntervalMethodFromExtremes.ci_mle,
# ConfidenceIntervalMethodFromExtremes.ci_bayes], # ConfidenceIntervalMethodFromExtremes.ci_bayes],
......
...@@ -116,17 +116,22 @@ class StudyVisualizerForNonStationaryTrends(StudyVisualizer): ...@@ -116,17 +116,22 @@ class StudyVisualizerForNonStationaryTrends(StudyVisualizer):
def plot_trends(self, max_abs_tdrl=None): def plot_trends(self, max_abs_tdrl=None):
if max_abs_tdrl is not None: if max_abs_tdrl is not None:
self.global_max_abs_tdrl = max_abs_tdrl self.global_max_abs_tdrl = max_abs_tdrl
self.study.visualize_study(massif_name_to_value=self.massif_name_to_tdrl_value, ax = self.study.visualize_study(massif_name_to_value=self.massif_name_to_tdrl_value,
replace_blue_by_white=False, axis_off=False, show_label=False, replace_blue_by_white=False,
axis_off=False, show_label=False,
add_colorbar=True, add_colorbar=True,
massif_name_to_marker_style=self.massif_name_to_marker_style, massif_name_to_marker_style=self.massif_name_to_marker_style,
massif_name_to_color=self.massif_name_to_tdrl_color, massif_name_to_color=self.massif_name_to_tdrl_color,
cmap=self.cmap, cmap=self.cmap,
show=self.show, show=False,
ticks_values_and_labels=self.ticks_values_and_labels, ticks_values_and_labels=self.ticks_values_and_labels,
label=self.label_tdrl_bar + ' for {}'.format(EUROCODE_RETURN_LEVEL_STR)) label=self.label_tdrl_bar + ' for {}'.format(EUROCODE_RETURN_LEVEL_STR))
ax.get_xaxis().set_visible(True)
ax.set_xticks([])
ax.set_xlabel('Altitude = {}m'.format(self.study.altitude), fontsize=12)
self.plot_name = 'tdlr_trends' self.plot_name = 'tdlr_trends'
self.show_or_save_to_file(add_classic_title=False, tight_layout=False, no_title=True) self.show_or_save_to_file(add_classic_title=False, tight_layout=True, no_title=True)
plt.close() plt.close()
@property @property
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment