create _adjust_axes_labels for axes labels

Serapieum-of-alex · Aug 26, 2024 · d41b29b · d41b29b
1 parent b21c07a
commit d41b29b
Showing 1 changed file with 32 additions and 12 deletions.
diff --git a/statista/time_series.py b/statista/time_series.py
@@ -119,6 +119,17 @@ def _get_ax_fig(
             fig = ax.figure
         return fig, ax
 
+    @staticmethod
+    def _adjust_axes_labels(
+        ax: Axes, tick_labels: List[str], title: str, xlabel: str, ylabel: str
+    ):
+        """Adjust the labels of the axes."""
+        ax.set_xticklabels(tick_labels)
+        ax.set_title(title)
+        ax.set_xlabel(xlabel)
+        ax.set_ylabel(ylabel)
+        return ax
+
     def box_plot(
         self, mean: bool = False, notch: bool = False, **kwargs
     ) -> Tuple[Figure, Axes]:
@@ -237,10 +248,13 @@ def box_plot(
                 )
             ),
         )
-        ax.set_xticklabels(self.columns)
-        ax.set_title(kwargs.get("title"))
-        ax.set_xlabel(kwargs.get("xlabel"))
-        ax.set_ylabel(kwargs.get("ylabel"))
+        ax = self._adjust_axes_labels(
+            ax,
+            self.columns,
+            kwargs.get("title"),
+            kwargs.get("xlabel"),
+            kwargs.get("ylabel"),
+        )
 
         ax.grid(kwargs.get("grid"), axis="both", linestyle="-.", linewidth=0.3)
         plt.show()
@@ -391,10 +405,13 @@ def violin(
             pc.set_alpha(color.get("alpha"))
 
         ax.xaxis.set_ticks(positions)
-        ax.set_xticklabels(self.columns)
-        ax.set_title(kwargs.get("title"))
-        ax.set_xlabel(kwargs.get("xlabel"))
-        ax.set_ylabel(kwargs.get("ylabel"))
+        ax = self._adjust_axes_labels(
+            ax,
+            self.columns,
+            kwargs.get("title"),
+            kwargs.get("xlabel"),
+            kwargs.get("ylabel"),
+        )
 
         ax.grid(kwargs.get("grid"), axis="both", linestyle="-.", linewidth=0.3)
         plt.show()
@@ -555,10 +572,13 @@ def raincloud(
 
         # Set x-tick labels
         ax.set_xticks(positions)
-        ax.set_xticklabels(self.columns)
-        ax.set_title(kwargs.get("title"))
-        ax.set_xlabel(kwargs.get("xlabel"))
-        ax.set_ylabel(kwargs.get("ylabel"))
+        ax = self._adjust_axes_labels(
+            ax,
+            self.columns,
+            kwargs.get("title"),
+            kwargs.get("xlabel"),
+            kwargs.get("ylabel"),
+        )
 
         ax.grid(kwargs.get("grid"), axis="both", linestyle="-.", linewidth=0.3)