Fix a bug that involves non-string columns

Solves issue #108  by converting all non-string color_groups to strings

Author: Jacobluke-

dabest/plotter.py

@@ -247,11 +247,13 @@ def effectsize_df_plotter(effectsize_df, **plot_kwargs):
     color_col = plot_kwargs["color_col"]
     if color_col is None:
         color_groups = pd.unique(plot_data[xvar])
+        color_groups = pd.Categorical([str(item) if isinstance(item, (int, float)) else item for item in color_groups])
         bootstraps_color_by_group = True
     else:
         if color_col not in plot_data.columns:
             raise KeyError("``{}`` is not a column in the data.".format(color_col))
         color_groups = pd.unique(plot_data[color_col])
+        color_groups = pd.Categorical([str(item) if isinstance(item, (int, float)) else item for item in color_groups])
         bootstraps_color_by_group = False
     if show_pairs:
         bootstraps_color_by_group = False

nbs/API/plotter.ipynb

@@ -306,11 +306,13 @@
     "    color_col = plot_kwargs[\"color_col\"]\n",
     "    if color_col is None:\n",
     "        color_groups = pd.unique(plot_data[xvar])\n",
+    "        color_groups = pd.Categorical([str(item) if isinstance(item, (int, float)) else item for item in color_groups])\n",
     "        bootstraps_color_by_group = True\n",
     "    else:\n",
     "        if color_col not in plot_data.columns:\n",
     "            raise KeyError(\"``{}`` is not a column in the data.\".format(color_col))\n",
     "        color_groups = pd.unique(plot_data[color_col])\n",
+    "        color_groups = pd.Categorical([str(item) if isinstance(item, (int, float)) else item for item in color_groups])\n",
     "        bootstraps_color_by_group = False\n",
     "    if show_pairs:\n",
     "        bootstraps_color_by_group = False\n",