diff --git a/experiment/meteo_france_data/visualization/hypercube_visualization/altitude_hypercube_visualizer.py b/experiment/meteo_france_data/visualization/hypercube_visualization/altitude_hypercube_visualizer.py index c216f6e28bc7e9036a36ae73d89c3b3a6e9d8438..3f1e44afce65d356b69d9ee57194ee9cc75f0781 100644 --- a/experiment/meteo_france_data/visualization/hypercube_visualization/altitude_hypercube_visualizer.py +++ b/experiment/meteo_france_data/visualization/hypercube_visualization/altitude_hypercube_visualizer.py @@ -25,7 +25,7 @@ class AltitudeHypercubeVisualizer(AbstractHypercubeVisualizer): @property def nb_axes(self): - return 2 + return 1 def trend_type_to_series(self, reduction_function): # Map each trend type to its serie with percentages @@ -72,6 +72,8 @@ class AltitudeHypercubeVisualizer(AbstractHypercubeVisualizer): subtitle=''): if axes is None: fig, axes = plt.subplots(self.nb_axes, 1, figsize=self.study_visualizer.figsize) + if not isinstance(axes, np.ndarray): + axes = [axes] trend_type_to_series = self.trend_type_to_series(reduction_function) for ax_idx, ax in enumerate(axes): @@ -168,13 +170,14 @@ class AltitudeHypercubeVisualizer(AbstractHypercubeVisualizer): class Altitude_Hypercube_Year_Visualizer(AltitudeHypercubeVisualizer): def get_title_plot(self, xlabel, ax_idx=None): - if ax_idx == 2: + if ax_idx == self.nb_axes - 1: return 'mean starting year' return super().get_title_plot(xlabel, ax_idx) @property def nb_axes(self): - return 3 + return super().nb_axes + 1 + @staticmethod def index_reduction(df, level): diff --git a/experiment/meteo_france_data/visualization/hypercube_visualization/main_hypercube_visualization.py b/experiment/meteo_france_data/visualization/hypercube_visualization/main_hypercube_visualization.py index 2669bf070f6cdf3a82a537b686ad33e0fcec792c..142ac101206fa4bac9fb191e58bb8313d188e8b2 100644 --- a/experiment/meteo_france_data/visualization/hypercube_visualization/main_hypercube_visualization.py +++ b/experiment/meteo_france_data/visualization/hypercube_visualization/main_hypercube_visualization.py @@ -54,7 +54,7 @@ def full_quantity_altitude_hypercube(): def fast_altitude_hypercube(): - save_to_file = True + save_to_file = False only_first_one = False fast = True altitudes = ALL_ALTITUDES[2:4] @@ -131,8 +131,8 @@ def fast_quantity_altitude_hypercube(): def main_run(): - fast_altitude_hypercube() - # fast_altitude_year_hypercube() + # fast_altitude_hypercube() + fast_altitude_year_hypercube() # full_altitude_year_hypercube() # fast_quantity_altitude_hypercube() # full_quantity_altitude_hypercube()