Adjusted default figure size for paired twogroup non proportional float_contrast plots (increase width a bit)

Author: JAnns98

dabest/misc_tools.py

@@ -685,7 +685,8 @@ def initialize_fig(
         effect_size_type: str, 
         yvar: str, 
         horizontal: bool, 
-        show_table: bool
+        show_table: bool,
+        color_col: str,
     ):
     """
     Initialize the figure and axes for the plotter function.
@@ -716,6 +717,8 @@ def initialize_fig(
         A boolean flag to determine if the plot is for horizontal plotting.
     show_table : dict
         A boolean flag to determine if the table will be shown in horizontal plot.
+    color_col : str
+        The column name for coloring the data points.
     """
     # Params
     fig_size = plot_kwargs["fig_size"]
@@ -735,7 +738,10 @@ def initialize_fig(
             fig_size = (7, 1 + (frac * all_groups_count))
         else:
             if is_paired and show_pairs and proportional is False:
-                frac = 0.8
+                if color_col is not None and float_contrast:
+                    frac = 0.9
+                else:
+                    frac = 0.8
             else:
                 frac = 1
             if float_contrast:

dabest/plotter.py

@@ -176,7 +176,8 @@ def effectsize_df_plotter(effectsize_df: object, **plot_kwargs) -> matplotlib.fi
                                                             effect_size_type = effect_size,
                                                             yvar = yvar,
                                                             horizontal = horizontal,
-                                                            show_table = table_kwargs['show']
+                                                            show_table = table_kwargs['show'],
+                                                            color_col = color_col
     )
 
     # Plotting the rawdata.

nbs/API/misc_tools.ipynb

@@ -739,7 +739,8 @@
     "        effect_size_type: str, \n",
     "        yvar: str, \n",
     "        horizontal: bool, \n",
-    "        show_table: bool\n",
+    "        show_table: bool,\n",
+    "        color_col: str,\n",
     "    ):\n",
     "    \"\"\"\n",
     "    Initialize the figure and axes for the plotter function.\n",
@@ -770,6 +771,8 @@
     "        A boolean flag to determine if the plot is for horizontal plotting.\n",
     "    show_table : dict\n",
     "        A boolean flag to determine if the table will be shown in horizontal plot.\n",
+    "    color_col : str\n",
+    "        The column name for coloring the data points.\n",
     "    \"\"\"\n",
     "    # Params\n",
     "    fig_size = plot_kwargs[\"fig_size\"]\n",
@@ -789,7 +792,10 @@
     "            fig_size = (7, 1 + (frac * all_groups_count))\n",
     "        else:\n",
     "            if is_paired and show_pairs and proportional is False:\n",
-    "                frac = 0.8\n",
+    "                if color_col is not None and float_contrast:\n",
+    "                    frac = 0.9\n",
+    "                else:\n",
+    "                    frac = 0.8\n",
     "            else:\n",
     "                frac = 1\n",
     "            if float_contrast:\n",

nbs/API/plotter.ipynb

@@ -233,7 +233,8 @@
     "                                                            effect_size_type = effect_size,\n",
     "                                                            yvar = yvar,\n",
     "                                                            horizontal = horizontal,\n",
-    "                                                            show_table = table_kwargs['show']\n",
+    "                                                            show_table = table_kwargs['show'],\n",
+    "                                                            color_col = color_col\n",
     "    )\n",
     "    \n",
     "    # Plotting the rawdata.\n",