py2ls 0.1.8.9__py3-none-any.whl → 0.1.9.1__py3-none-any.whl

py2ls/plot.py

@@ -3,12 +3,13 @@ import numpy as np
 import pandas as pd
 from matplotlib.colors import to_rgba
 from scipy.stats import gaussian_kde
-
+import seaborn as sns
 import matplotlib
 import matplotlib.ticker as tck
 from cycler import cycler
 import logging
-from .ips import fsave
+import os
+from .ips import fsave, fload, mkdir
 
 # Suppress INFO messages from fontTools
 logging.getLogger("fontTools").setLevel(logging.WARNING)
@@ -17,7 +18,10 @@ logging.getLogger("fontTools").setLevel(logging.WARNING)
 def catplot(data, *args, **kwargs):
     """
     catplot(data, opt=None, ax=None)
-
+    The catplot function is designed to provide a flexible way to create various types of
+    categorical plots. It supports multiple plot layers such as bars, error bars, scatter
+    plots, box plots, violin plots, and lines. Each plot type is handled by its own internal
+    function, allowing for separation of concerns and modularity in the design.
     Args:
         data (array): data matrix
     """
@@ -459,255 +463,296 @@ def catplot(data, *args, **kwargs):
         # custom_order = ['s', 'bx', 'e']
         # full_order = sort_catplot_layers(custom_order)
 
-    # figsets
-    kw_figsets = kwargs.get("figsets", None)
-    # check the data type
-    if isinstance(data, pd.DataFrame):
-        df = data.copy()
-        x = kwargs.get("x", None)
-        y = kwargs.get("y", None)
-        hue = kwargs.get("hue", None)
-        data = df2array(data=data, x=x, y=y, hue=hue)
-        xticklabels = []
-        if hue is not None:
-            for i in df[x].unique().tolist():
-                for j in df[hue].unique().tolist():
-                    xticklabels.append(i + "-" + j)
-            x_len = len(df[x].unique().tolist())
-            hue_len = len(df[hue].unique().tolist())
-            xticks = generate_xticks_with_gap(x_len, hue_len)
-            default_x_width = 0.85
-            legend_hue = df[hue].unique().tolist()
-            default_colors = get_color(hue_len)
+    col = kwargs.get("col", None)
+    if not col:
+        # figsets
+        kw_figsets = kwargs.get("figsets", None)
+        # check the data type
+        if isinstance(data, pd.DataFrame):
+            df = data.copy()
+            x = kwargs.get("x", None)
+            y = kwargs.get("y", None)
+            hue = kwargs.get("hue", None)
+            data = df2array(data=data, x=x, y=y, hue=hue)
+            xticklabels = []
+            if hue is not None:
+                # for i in df[x].unique().tolist():
+                #     for j in df[hue].unique().tolist():
+                #         xticklabels.append(i + "-" + j)
+                for i in df[x].unique().tolist():
+                    xticklabels.append(i)
+                x_len = len(df[x].unique().tolist())
+                hue_len = len(df[hue].unique().tolist())
+                xticks = generate_xticks_with_gap(x_len, hue_len)
+                xticks_x_loc = generate_xticks_x_labels(x_len, hue_len)
+                default_x_width = 0.85
+                legend_hue = df[hue].unique().tolist()
+                default_colors = get_color(hue_len)
+            else:
+                for i in df[x].unique().tolist():
+                    xticklabels.append(i)
+                xticks = np.arange(1, len(xticklabels) + 1).tolist()
+                xticks_x_loc = np.arange(1, len(xticklabels) + 1).tolist()
+                legend_hue = xticklabels
+                default_colors = get_color(len(xticklabels))
+                default_x_width = 0.5
+
+            # when the xticklabels are too long, rotate the labels a bit
+
+            xangle = 30 if max([len(i) for i in xticklabels]) > 50 else 0
+            if kw_figsets is not None:
+                kw_figsets = {
+                    "ylabel": y,
+                    # "xlabel": x,
+                    "xticks": xticks_x_loc,  # xticks,
+                    "xticklabels": xticklabels,
+                    "xangle": xangle,
+                    **kw_figsets,
+                }
+            else:
+                kw_figsets = {
+                    "ylabel": y,
+                    # "xlabel": x,
+                    "xticks": xticks_x_loc,  # xticks,
+                    "xticklabels": xticklabels,
+                    "xangle": xangle,
+                }
         else:
-            for i in df[x].unique().tolist():
-                xticklabels.append(i)
-            xticks = np.arange(1, len(xticklabels) + 1).tolist()
-            legend_hue = xticklabels
-            default_colors = get_color(len(xticklabels))
+            xticks = np.arange(1, data.shape[1] + 1).tolist()
             default_x_width = 0.5
-
-        # when the xticklabels are too long, rotate the labels a bit
-
-        xangle = 30 if max([len(i) for i in xticklabels]) > 5 else 0
-        if kw_figsets is not None:
-            kw_figsets = {
-                "ylabel": y,
-                "xlabel": x,
-                "xticks": xticks,
-                "xticklabels": xticklabels,
-                "xangle": xangle,
-                **kw_figsets,
-            }
+            default_colors = get_color(len(xticks))
+            legend_hue = None
+            xangle = 0
+
+        # full_order
+        opt = kwargs.get("opt", {})
+        ax = kwargs.get("ax", None)
+        if "ax" not in locals() or ax is None:
+            ax = plt.gca()
+        opt.setdefault("c", default_colors)
+        # if len(opt["c"]) < data.shape[1]:
+        #     additional_colors = plt.cm.winter(
+        #         np.linspace(0, 1, data.shape[1] - len(opt["c"]))
+        #     )
+        #     opt["c"] = np.vstack([opt["c"], additional_colors[:, :3]])
+
+        opt.setdefault("loc", {})
+        opt["loc"].setdefault("go", 0)
+        opt["loc"].setdefault("xloc", xticks)
+
+        # export setting
+        opt.setdefault("style", {})
+        opt["style"].setdefault("export", None)
+        print(opt["style"])
+
+        # opt.setdefault('layer', {})
+        opt.setdefault("layer", ["b", "bx", "e", "v", "s", "l"])
+
+        opt.setdefault("b", {})
+        opt["b"].setdefault("go", 1)
+        opt["b"].setdefault("loc", "c")
+        opt["b"].setdefault("FaceColor", opt["c"])
+        opt["b"].setdefault("FaceAlpha", 1)
+        opt["b"].setdefault("EdgeColor", "k")
+        opt["b"].setdefault("EdgeAlpha", 1)
+        opt["b"].setdefault("LineStyle", "-")
+        opt["b"].setdefault("LineWidth", 0.8)
+        opt["b"].setdefault("x_width", default_x_width)
+        opt["b"].setdefault("ShowBaseLine", "off")
+        opt["b"].setdefault("hatch", None)
+
+        opt.setdefault("e", {})
+        opt["e"].setdefault("go", 1)
+        opt["e"].setdefault("loc", "l")
+        opt["e"].setdefault("LineWidth", 1)
+        opt["e"].setdefault("CapLineWidth", 1)
+        opt["e"].setdefault("CapSize", 2)
+        opt["e"].setdefault("Marker", "none")
+        opt["e"].setdefault("LineStyle", "none")
+        opt["e"].setdefault("LineColor", "k")
+        opt["e"].setdefault("LineJoin", "round")
+        opt["e"].setdefault("MarkerSize", "auto")
+        opt["e"].setdefault("FaceColor", opt["c"])
+        opt["e"].setdefault("MarkerEdgeColor", "none")
+        opt["e"].setdefault("Visible", True)
+        opt["e"].setdefault("Orientation", "vertical")
+        opt["e"].setdefault("error", "sem")
+        opt["e"].setdefault("x_width", default_x_width / 5)
+        opt["e"].setdefault("cap_dir", "b")
+
+        opt.setdefault("s", {})
+        opt["s"].setdefault("go", 1)
+        opt["s"].setdefault("loc", "r")
+        opt["s"].setdefault("FaceColor", "w")
+        opt["s"].setdefault("cmap", None)
+        opt["s"].setdefault("FaceAlpha", 1)
+        opt["s"].setdefault("x_width", default_x_width / 5)
+        opt["s"].setdefault("Marker", "o")
+        opt["s"].setdefault("MarkerSize", 15)
+        opt["s"].setdefault("LineWidth", 0.8)
+        opt["s"].setdefault("MarkerEdgeColor", "k")
+
+        opt.setdefault("l", {})
+        opt["l"].setdefault("go", 1)
+        opt["l"].setdefault("LineStyle", "-")
+        opt["l"].setdefault("LineColor", "k")
+        opt["l"].setdefault("LineWidth", 0.5)
+        opt["l"].setdefault("LineAlpha", 0.5)
+
+        opt.setdefault("bx", {})
+        opt["bx"].setdefault("go", 0)
+        opt["bx"].setdefault("loc", "r")
+        opt["bx"].setdefault("FaceColor", opt["c"])
+        opt["bx"].setdefault("EdgeColor", "k")
+        opt["bx"].setdefault("FaceAlpha", 0.85)
+        opt["bx"].setdefault("EdgeAlpha", 1)
+        opt["bx"].setdefault("LineStyle", "-")
+        opt["bx"].setdefault("x_width", default_x_width / 5)
+        opt["bx"].setdefault("ShowBaseLine", "off")
+        opt["bx"].setdefault("Notch", False)
+        opt["bx"].setdefault("Outliers", "on")
+        opt["bx"].setdefault("OutlierMarker", "+")
+        opt["bx"].setdefault("OutlierColor", "r")
+        opt["bx"].setdefault("OutlierSize", 6)
+        # opt['bx'].setdefault('PlotStyle', 'traditional')
+        # opt['bx'].setdefault('FactorDirection', 'auto')
+        opt["bx"].setdefault("LineWidth", 0.5)
+        opt["bx"].setdefault("Whisker", opt["bx"]["LineWidth"])
+        opt["bx"].setdefault("Orientation", "vertical")
+        opt["bx"].setdefault("BoxLineWidth", opt["bx"]["LineWidth"])
+        opt["bx"].setdefault("FaceColor", "k")
+        opt["bx"].setdefault("WhiskerLineStyle", "-")
+        opt["bx"].setdefault("WhiskerLineColor", "k")
+        opt["bx"].setdefault("WhiskerLineWidth", opt["bx"]["LineWidth"])
+        opt["bx"].setdefault("Caps", True)
+        opt["bx"].setdefault("CapLineColor", "k")
+        opt["bx"].setdefault("CapLineWidth", opt["bx"]["LineWidth"])
+        opt["bx"].setdefault("CapSize", 0.2)
+        opt["bx"].setdefault("MedianLine", True)
+        opt["bx"].setdefault("MedianLineStyle", "-")
+        opt["bx"].setdefault("MedianStyle", "line")
+        opt["bx"].setdefault("MedianLineColor", "k")
+        opt["bx"].setdefault("MedianLineWidth", opt["bx"]["LineWidth"] * 4)
+        opt["bx"].setdefault("MedianLineTop", False)
+        opt["bx"].setdefault("MeanLine", False)
+        opt["bx"].setdefault("showmeans", opt["bx"]["MeanLine"])
+        opt["bx"].setdefault("MeanLineStyle", "-")
+        opt["bx"].setdefault("MeanLineColor", "w")
+        opt["bx"].setdefault("MeanLineWidth", opt["bx"]["LineWidth"] * 4)
+
+        # Violin plot options
+        opt.setdefault("v", {})
+        opt["v"].setdefault("go", 0)
+        opt["v"].setdefault("x_width", 0.3)
+        opt["v"].setdefault("loc", "r")
+        opt["v"].setdefault("EdgeColor", "none")
+        opt["v"].setdefault("FaceColor", opt["c"])
+        opt["v"].setdefault("FaceAlpha", 0.3)
+        opt["v"].setdefault("BandWidth", "scott")
+        opt["v"].setdefault("Function", "pdf")
+        opt["v"].setdefault("Kernel", "gau")
+        opt["v"].setdefault("NumPoints", 500)
+        opt["v"].setdefault("BoundaryCorrection", "reflection")
+
+        # load style:
+        style_use = kwargs.get("style_use", None)
+        if style_use:
+            try:
+                dir_curr_script = os.path.dirname(os.path.abspath(__file__))
+                dir_style = dir_curr_script + "/data/styles/"
+                style_load = fload(dir_style + style_use + ".json")
+                opt.update(style_load)
+            except:
+                print(f"cannot find the style'{style_name}'")
+        data_m = np.nanmean(data, axis=0)
+        nr, nc = data.shape
+
+        for key in kwargs.keys():
+            if key in opt:
+                if isinstance(kwargs[key], dict):
+                    opt[key].update(kwargs[key])
+                else:
+                    opt[key] = kwargs[key]
+        if isinstance(opt["loc"]["xloc"], list):
+            xloc = np.array(opt["loc"]["xloc"])
         else:
-            kw_figsets = {
-                "ylabel": y,
-                "xlabel": x,
-                "xticks": xticks,
-                "xticklabels": xticklabels,
-                "xangle": xangle,
-            }
-    else:
-        xticks = np.arange(1, data.shape[1] + 1).tolist()
-        default_x_width = 0.5
-        xangle = 0
+            xloc = opt["loc"]["xloc"]
+        layers = sort_catplot_layers(opt["layer"])
+
+        label_which = kwargs.get("label_which", "barplot")
+        if "b" in label_which:
+            legend_which = "b"
+        elif "s" in label_which:
+            legend_which = "s"
+        elif "bx" in label_which:
+            legend_which = "bx"
+        elif "e" in label_which:
+            legend_which = "e"
+        elif "v" in label_which:
+            legend_which = "v"
+        else:
+            legend_which = None
 
-    # full_order
-    opt = kwargs.get("opt", {})
-    ax = kwargs.get("ax", None)
-    if "ax" not in locals() or ax is None:
-        ax = plt.gca()
-    opt.setdefault("c", default_colors)
-    # if len(opt["c"]) < data.shape[1]:
-    #     additional_colors = plt.cm.winter(
-    #         np.linspace(0, 1, data.shape[1] - len(opt["c"]))
-    #     )
-    #     opt["c"] = np.vstack([opt["c"], additional_colors[:, :3]])
-
-    opt.setdefault("loc", {})
-    opt["loc"].setdefault("go", 0)
-    opt["loc"].setdefault("xloc", xticks)
-
-    # export setting
-    opt.setdefault("export", {})
-    opt["export"].setdefault("path", None)
-    print(opt["export"])
-
-    # opt.setdefault('layer', {})
-    opt.setdefault("layer", ["b", "bx", "e", "v", "s", "l"])
-
-    opt.setdefault("b", {})
-    opt["b"].setdefault("go", 1)
-    opt["b"].setdefault("loc", "c")
-    opt["b"].setdefault("FaceColor", opt["c"])
-    opt["b"].setdefault("FaceAlpha", 1)
-    opt["b"].setdefault("EdgeColor", "k")
-    opt["b"].setdefault("EdgeAlpha", 1)
-    opt["b"].setdefault("LineStyle", "-")
-    opt["b"].setdefault("LineWidth", 0.8)
-    opt["b"].setdefault("x_width", default_x_width)
-    opt["b"].setdefault("ShowBaseLine", "off")
-    opt["b"].setdefault("hatch", None)
-
-    opt.setdefault("e", {})
-    opt["e"].setdefault("go", 1)
-    opt["e"].setdefault("loc", "l")
-    opt["e"].setdefault("LineWidth", 1)
-    opt["e"].setdefault("CapLineWidth", 1)
-    opt["e"].setdefault("CapSize", 2)
-    opt["e"].setdefault("Marker", "none")
-    opt["e"].setdefault("LineStyle", "none")
-    opt["e"].setdefault("LineColor", "k")
-    opt["e"].setdefault("LineJoin", "round")
-    opt["e"].setdefault("MarkerSize", "auto")
-    opt["e"].setdefault("FaceColor", opt["c"])
-    opt["e"].setdefault("MarkerEdgeColor", "none")
-    opt["e"].setdefault("Visible", True)
-    opt["e"].setdefault("Orientation", "vertical")
-    opt["e"].setdefault("error", "sem")
-    opt["e"].setdefault("x_width", opt["b"]["x_width"] / 5)
-    opt["e"].setdefault("cap_dir", "b")
-
-    opt.setdefault("s", {})
-    opt["s"].setdefault("go", 1)
-    opt["s"].setdefault("loc", "r")
-    opt["s"].setdefault("FaceColor", "w")
-    opt["s"].setdefault("cmap", None)
-    opt["s"].setdefault("FaceAlpha", 1)
-    opt["s"].setdefault("x_width", opt["b"]["x_width"] / 5)
-    opt["s"].setdefault("Marker", "o")
-    opt["s"].setdefault("MarkerSize", 15)
-    opt["s"].setdefault("LineWidth", 0.8)
-    opt["s"].setdefault("MarkerEdgeColor", "k")
-
-    opt.setdefault("l", {})
-    opt["l"].setdefault("go", 1)
-    opt["l"].setdefault("LineStyle", "-")
-    opt["l"].setdefault("LineColor", "k")
-    opt["l"].setdefault("LineWidth", 0.5)
-    opt["l"].setdefault("LineAlpha", 0.5)
-
-    opt.setdefault("bx", {})
-    opt["bx"].setdefault("go", 0)
-    opt["bx"].setdefault("loc", "r")
-    opt["bx"].setdefault("FaceColor", opt["c"])
-    opt["bx"].setdefault("EdgeColor", "k")
-    opt["bx"].setdefault("FaceAlpha", 0.85)
-    opt["bx"].setdefault("EdgeAlpha", 1)
-    opt["bx"].setdefault("LineStyle", "-")
-    opt["bx"].setdefault("x_width", 0.2)
-    opt["bx"].setdefault("ShowBaseLine", "off")
-    opt["bx"].setdefault("Notch", False)
-    opt["bx"].setdefault("Outliers", "on")
-    opt["bx"].setdefault("OutlierMarker", "+")
-    opt["bx"].setdefault("OutlierColor", "r")
-    opt["bx"].setdefault("OutlierSize", 6)
-    # opt['bx'].setdefault('PlotStyle', 'traditional')
-    # opt['bx'].setdefault('FactorDirection', 'auto')
-    opt["bx"].setdefault("Whisker", 0.5)
-    opt["bx"].setdefault("Orientation", "vertical")
-    opt["bx"].setdefault("BoxLineWidth", 0.5)
-    opt["bx"].setdefault("FaceColor", "k")
-    opt["bx"].setdefault("WhiskerLineStyle", "-")
-    opt["bx"].setdefault("WhiskerLineColor", "k")
-    opt["bx"].setdefault("WhiskerLineWidth", 0.5)
-    opt["bx"].setdefault("Caps", True)
-    opt["bx"].setdefault("CapLineColor", "k")
-    opt["bx"].setdefault("CapLineWidth", 0.5)
-    opt["bx"].setdefault("CapSize", 0.2)
-    opt["bx"].setdefault("MedianLine", True)
-    opt["bx"].setdefault("MedianLineStyle", "-")
-    opt["bx"].setdefault("MedianStyle", "line")
-    opt["bx"].setdefault("MedianLineColor", "k")
-    opt["bx"].setdefault("MedianLineWidth", 2)
-    opt["bx"].setdefault("MedianLineTop", False)
-    opt["bx"].setdefault("MeanLine", False)
-    opt["bx"].setdefault("showmeans", opt["bx"]["MeanLine"])
-    opt["bx"].setdefault("MeanLineStyle", "-")
-    opt["bx"].setdefault("MeanLineColor", "w")
-    opt["bx"].setdefault("MeanLineWidth", 2)
-
-    # Violin plot options
-    opt.setdefault("v", {})
-    opt["v"].setdefault("go", 0)
-    opt["v"].setdefault("x_width", 0.3)
-    opt["v"].setdefault("loc", "r")
-    opt["v"].setdefault("EdgeColor", "none")
-    opt["v"].setdefault("FaceColor", opt["c"])
-    opt["v"].setdefault("FaceAlpha", 0.3)
-    opt["v"].setdefault("BandWidth", "scott")
-    opt["v"].setdefault("Function", "pdf")
-    opt["v"].setdefault("Kernel", "gau")
-    opt["v"].setdefault("NumPoints", 500)
-    opt["v"].setdefault("BoundaryCorrection", "reflection")
-
-    data_m = np.nanmean(data, axis=0)
-    nr, nc = data.shape
-
-    for key in kwargs.keys():
-        if key in opt:
-            if isinstance(kwargs[key], dict):
-                opt[key].update(kwargs[key])
+        for layer in layers:
+            if layer == "b" and opt["b"]["go"]:
+                if legend_which == "b":
+                    plot_bars(data, data_m, opt["b"], xloc, ax, label=legend_hue)
+                else:
+                    plot_bars(data, data_m, opt["b"], xloc, ax, label=None)
+            elif layer == "e" and opt["e"]["go"]:
+                if legend_which == "e":
+                    plot_errors(data, data_m, opt["e"], xloc, ax, label=legend_hue)
+                else:
+                    plot_errors(data, data_m, opt["e"], xloc, ax, label=None)
+            elif layer == "s" and opt["s"]["go"]:
+                if legend_which == "s":
+                    plot_scatter(data, opt["s"], xloc, ax, label=legend_hue)
+                else:
+                    plot_scatter(data, opt["s"], xloc, ax, label=None)
+            elif layer == "bx" and opt["bx"]["go"]:
+                if legend_which == "bx":
+                    plot_boxplot(data, opt["bx"], xloc, ax, label=legend_hue)
+                else:
+                    plot_boxplot(data, opt["bx"], xloc, ax, label=None)
+            elif layer == "v" and opt["v"]["go"]:
+                if legend_which == "v":
+                    plot_violin(data, opt["v"], xloc, ax, label=legend_hue)
+                else:
+                    plot_violin(data, opt["v"], xloc, ax, label=None)
+            elif all([layer == "l", opt["l"]["go"], opt["s"]["go"]]):
+                plot_lines(data, opt["l"], opt["s"], ax)
             else:
-                opt[key] = kwargs[key]
-    if isinstance(opt["loc"]["xloc"], list):
-        xloc = np.array(opt["loc"]["xloc"])
-    else:
-        xloc = opt["loc"]["xloc"]
-    layers = sort_catplot_layers(opt["layer"])
-
-    label_which = kwargs.get("label_which", "barplot")
-    if "b" in label_which:
-        legend_which = "b"
-    elif "s" in label_which:
-        legend_which = "s"
-    elif "bx" in label_which:
-        legend_which = "bx"
-    elif "e" in label_which:
-        legend_which = "e"
-    elif "v" in label_which:
-        legend_which = "v"
+                print("layers run some problems")
+        if kw_figsets is not None:
+            figsets(ax=ax, **kw_figsets)
+        show_legend = kwargs.get("show_legend", True)
+        if show_legend:
+            ax.legend()
+
+        style_export = kwargs.get("style_export", None)
+        if style_export and (style_export != style_use):
+            dir_curr_script = os.path.dirname(os.path.abspath(__file__))
+            dir_style = dir_curr_script + "/data/styles/"
+            fsave(dir_style + style_export + ".json", opt)
+        return ax, opt
     else:
-        legend_which = None
-
-    for layer in layers:
-        if layer == "b" and opt["b"]["go"]:
-            if legend_which == "b":
-                plot_bars(data, data_m, opt["b"], xloc, ax, label=legend_hue)
-            else:
-                plot_bars(data, data_m, opt["b"], xloc, ax, label=None)
-        elif layer == "e" and opt["e"]["go"]:
-            if legend_which == "e":
-                plot_errors(data, data_m, opt["e"], xloc, ax, label=legend_hue)
-            else:
-                plot_errors(data, data_m, opt["e"], xloc, ax, label=None)
-        elif layer == "s" and opt["s"]["go"]:
-            if legend_which == "s":
-                plot_scatter(data, opt["s"], xloc, ax, label=legend_hue)
-            else:
-                plot_scatter(data, opt["s"], xloc, ax, label=None)
-        elif layer == "bx" and opt["bx"]["go"]:
-            if legend_which == "bx":
-                plot_boxplot(data, opt["bx"], xloc, ax, label=legend_hue)
-            else:
-                plot_boxplot(data, opt["bx"], xloc, ax, label=None)
-        elif layer == "v" and opt["v"]["go"]:
-            if legend_which == "v":
-                plot_violin(data, opt["v"], xloc, ax, label=legend_hue)
-            else:
-                plot_violin(data, opt["v"], xloc, ax, label=None)
-        elif all([layer == "l", opt["l"]["go"], opt["s"]["go"]]):
-            plot_lines(data, opt["l"], opt["s"], ax)
-
-    if kw_figsets is not None:
-        figsets(ax=ax, **kw_figsets)
-    show_legend = kwargs.get("show_legend", True)
-    if show_legend:
-        ax.legend()
-    export_style = kwargs.get("export_style", None)
-    if export_style:
-        fsave(export_style, opt)
-    return ax, opt
+        col_names = data[col].unique().tolist()
+        nrow, ncol = kwargs.get("subplots", [len(col_names), 1])
+        figsize = kwargs.get("figsize", [3 * ncol, 3 * nrow])
+        fig, axs = plt.subplots(nrow, ncol, figsize=figsize, squeeze=False)
+        axs = axs.flatten()
+        key2rm = ["data", "ax", "col", "subplots"]
+        for k2rm in key2rm:
+            if k2rm in kwargs:
+                del kwargs[k2rm]
+        for i, ax in enumerate(axs):
+            # ax = axs[i][0] if len(col_names) > 1 else axs[0]
+            if i < len(col_names):
+                df_sub = data.loc[data[col] == col_names[i]]
+                _, opt = catplot(ax=ax, data=df_sub, **kwargs)
+                ax.set_title(col_names[i])
+        print(f"Axis layout shape: {axs.shape}")
+        return axs, opt
 
 
 def get_cmap():
@@ -1675,3 +1720,11 @@ def generate_xticks_with_gap(x_len, hue_len):
     positive_array = concatenated_array[concatenated_array > 0].tolist()
 
     return positive_array
+
+
+def generate_xticks_x_labels(x_len, hue_len):
+    arrays = [
+        np.arange(1, hue_len + 1) + hue_len * (x_len - i) + (x_len - i)
+        for i in range(max(x_len, hue_len), 0, -1)  # i iterates from 3 to 1
+    ]
+    return [np.mean(i) for i in arrays if np.mean(i) > 0]

{py2ls-0.1.8.9.dist-info → py2ls-0.1.9.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: py2ls
-Version: 0.1.8.9
+Version: 0.1.9.1
 Summary: py(thon)2(too)ls
 Author: Jianfeng
 Author-email: Jianfeng.Liu0413@gmail.com

{py2ls-0.1.8.9.dist-info → py2ls-0.1.9.1.dist-info}/RECORD

@@ -126,22 +126,28 @@ py2ls/__init__.py,sha256=Nn8jTIvySX7t7DMJ8VNRVctTStgXGjHldOIdZ35PdW8,165
 py2ls/brain_atlas.py,sha256=w1o5EelRjq89zuFJUNSz4Da8HnTCwAwDAZ4NU4a-bAY,5486
 py2ls/chat.py,sha256=Yr22GoIvoWhpV3m4fdwV_I0Mn77La346_ymSinR-ORA,3793
 py2ls/correlators.py,sha256=RbOaJIPLCHJtUm5SFi_4dCJ7VFUPWR0PErfK3K26ad4,18243
-py2ls/data/.DS_Store,sha256=1lFlJ5EFymdzGAUAaI30vcaaLHt3F1LwpG7xILf9jsM,6148
+py2ls/data/.DS_Store,sha256=iH2O541jT_5mlTPavY_d5V2prS9zhNx4Pv7yhmbwaHI,6148
 py2ls/data/db2ls_sql_chtsht.json,sha256=ls9d7Sm8TLeujanWHfHlWhU85Qz1KnAizO_9X3wUH7E,6933
 py2ls/data/docs_links.json,sha256=kXgbbWo0b8bfV4n6iuuUNLnZipIyLzokUO6Lzmf7nO4,101829
 py2ls/data/lang_code_iso639.json,sha256=qZiU7H2RLJjDMXK22C-jhwzLJCI5vKmampjB1ys4ek4,2157
+py2ls/data/styles/style1.json,sha256=Q3tdH0Sf08FjNUZE5mELA45JEw3BXjSAL2nLfFDn1bU,3101
+py2ls/data/styles/style2.json,sha256=2xhDv-_qQOKaODy8fWRoaQk_W5-I3EdA6uh4JNnINGg,3124
+py2ls/data/styles/style3.json,sha256=0lHmjFGqlf1c7HLllsgGVNFkuEsqSCicBv-iOTB9hRk,3126
+py2ls/data/styles/style4.json,sha256=G8thPHwmJyS3kDletrh3NkapZ03bNfey2-zpG4erBfk,3072
+py2ls/data/styles/style5.json,sha256=0bqt3CYM1iBtu_7D8LmurnZ2mlrw-zOdUMUpnUADih4,3069
+py2ls/data/styles/style6.json,sha256=0bqt3CYM1iBtu_7D8LmurnZ2mlrw-zOdUMUpnUADih4,3069
 py2ls/db2ls.py,sha256=MMfFX47aIPIyu7fU9aPvX9lbPRPYOpJ_VXwlnWk-8qo,13615
 py2ls/doc.py,sha256=xN3g1OWfoaGUhikbJ0NqbN5eKy1VZVvWwRlhHMgyVEc,4243
 py2ls/export_requirements.py,sha256=x2WgUF0jYKz9GfA1MVKN-MdsM-oQ8yUeC6Ua8oCymio,2325
 py2ls/freqanalysis.py,sha256=F4218VSPbgL5tnngh6xNCYuNnfR-F_QjECUUxrPYZss,32594
 py2ls/ips.py,sha256=N7MdOCgJXDQu73YkJQTtDN3RSntzXX7V0MOJ1NYBLEk,100572
 py2ls/netfinder.py,sha256=OMStrwMAASf1ajlyEfseoaEygo7G5WKBAFRE0LY15Lw,49477
-py2ls/plot.py,sha256=9_vyFNoPX4c8b0AS3lyFu18QKec5je3f0G8T7di9K9o,63197
+py2ls/plot.py,sha256=kR8NcWUnzqCEi0r8ZPOaOtk-S5JUuLpjq7wsWSf39wc,66893
 py2ls/setuptools-70.1.0-py3-none-any.whl,sha256=2bi3cUVal8ip86s0SOvgspteEF8SKLukECi-EWmFomc,882588
 py2ls/sleep_events_detectors.py,sha256=bQA3HJqv5qnYKJJEIhCyhlDtkXQfIzqksnD0YRXso68,52145
 py2ls/stats.py,sha256=Wd9yCKQ_61QD29WMEgMuEcreFxF91NmlPW65iWT2B5w,39041
 py2ls/translator.py,sha256=bc5FB-wqC4TtQz9gyCP1mE38HqNRJ_pmuRIgKnAlMzM,30581
 py2ls/wb_detector.py,sha256=7y6TmBUj9exCZeIgBAJ_9hwuhkDh1x_-yg4dvNY1_GQ,6284
-py2ls-0.1.8.9.dist-info/METADATA,sha256=7-nSwoDVg7Yf3lZ8pqIZ6b-voypM-2p9vwDWoSLqGFs,20017
-py2ls-0.1.8.9.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-py2ls-0.1.8.9.dist-info/RECORD,,
+py2ls-0.1.9.1.dist-info/METADATA,sha256=XToL4jUZomQ3-3jUt-v5uSY8pFwD6YQu_qfnxLtvAog,20017
+py2ls-0.1.9.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+py2ls-0.1.9.1.dist-info/RECORD,,