py2ls 0.1.4.9__py3-none-any.whl → 0.1.5.0__py3-none-any.whl

py2ls/stdshade.py

@@ -0,0 +1,173 @@
+from scipy.signal import savgol_filter
+import numpy as np
+import matplotlib.pyplot as plt
+
+def stdshade(ax=None,*args, **kwargs):
+
+    def hue2rgb(hex_colors):
+        def hex_to_rgb(hex_color):
+            """Converts a hexadecimal color code to RGB values."""
+            if hex_colors.startswith("#"):
+                hex_color = hex_color.lstrip("#")
+            return tuple(int(hex_color[i : i + 2], 16) / 255.0 for i in (0, 2, 4))
+        if isinstance(hex_colors, str):
+            return hex_to_rgb(hex_colors)
+        elif isinstance(hex_colors, (list)):
+            """Converts a list of hexadecimal color codes to a list of RGB values."""
+            rgb_values = [hex_to_rgb(hex_color) for hex_color in hex_colors]
+            return rgb_values
+    if (
+        isinstance(ax, np.ndarray)
+        and ax.ndim == 2
+        and min(ax.shape) > 1
+        and max(ax.shape) > 1
+    ):
+        y = ax
+        ax = plt.gca()
+    if ax is None:
+        ax = plt.gca()
+    alpha = 0.5
+    acolor = "k"
+    paraStdSem = "sem"
+    plotStyle = "-"
+    plotMarker = "none"
+    smth = 1
+    l_c_one = ["r", "g", "b", "m", "c", "y", "k", "w"]
+    l_style2 = ["--", "-."]
+    l_style1 = ["-", ":"]
+    l_mark = ["o", "+", "*", ".", "x", "_", "|", "s", "d", "^", "v", ">", "<", "p", "h"]
+    # Check each argument
+    for iarg in range(len(args)):
+        if (
+            isinstance(args[iarg], np.ndarray)
+            and args[iarg].ndim == 2
+            and min(args[iarg].shape) > 1
+            and max(args[iarg].shape) > 1
+        ):
+            y = args[iarg]
+        # Except y, continuous data is 'F'
+        if (isinstance(args[iarg], np.ndarray) and args[iarg].ndim == 1) or isinstance(
+            args[iarg], range
+        ):
+            x = args[iarg]
+            if isinstance(x, range):
+                x = np.arange(start=x.start, stop=x.stop, step=x.step)
+        # Only one number( 0~1), 'alpha' / color
+        if isinstance(args[iarg], (int, float)):
+            if np.size(args[iarg]) == 1 and 0 <= args[iarg] <= 1:
+                alpha = args[iarg]
+        if isinstance(args[iarg], (list, tuple)) and np.size(args[iarg]) == 3:
+            acolor = args[iarg]
+            acolor = tuple(acolor) if isinstance(acolor, list) else acolor
+        # Color / plotStyle /
+        if (
+            isinstance(args[iarg], str)
+            and len(args[iarg]) == 1
+            and args[iarg] in l_c_one
+        ):
+            acolor = args[iarg]
+        else:
+            if isinstance(args[iarg], str):
+                if args[iarg] in ["sem", "std"]:
+                    paraStdSem = args[iarg]
+                if args[iarg].startswith("#"):
+                    acolor=hue2rgb(args[iarg])
+                if str2list(args[iarg])[0] in l_c_one:
+                    if len(args[iarg]) == 3:
+                        k = [i for i in str2list(args[iarg]) if i in l_c_one]
+                        if k != []:
+                            acolor = k[0] 
+                        st = [i for i in l_style2 if i in args[iarg]]
+                        if st != []:
+                            plotStyle = st[0] 
+                    elif len(args[iarg]) == 2:
+                        k = [i for i in str2list(args[iarg]) if i in l_c_one]
+                        if k != []:
+                            acolor = k[0] 
+                        mk = [i for i in str2list(args[iarg]) if i in l_mark]
+                        if mk != []:
+                            plotMarker = mk[0] 
+                        st = [i for i in l_style1 if i in args[iarg]]
+                        if st != []:
+                            plotStyle = st[0] 
+                if len(args[iarg]) == 1:
+                    k = [i for i in str2list(args[iarg]) if i in l_c_one]
+                    if k != []:
+                        acolor = k[0] 
+                    mk = [i for i in str2list(args[iarg]) if i in l_mark]
+                    if mk != []:
+                        plotMarker = mk[0] 
+                    st = [i for i in l_style1 if i in args[iarg]]
+                    if st != []:
+                        plotStyle = st[0] 
+                if len(args[iarg]) == 2:
+                    st = [i for i in l_style2 if i in args[iarg]]
+                    if st != []:
+                        plotStyle = st[0]
+        # smth
+        if (
+            isinstance(args[iarg], (int, float))
+            and np.size(args[iarg]) == 1
+            and args[iarg] >= 1
+        ):
+            smth = args[iarg]
+
+    if "x" not in locals() or x is None:
+        x = np.arange(1, y.shape[1] + 1)
+    elif len(x) < y.shape[1]:
+        y = y[:, x]
+        nRow = y.shape[0]
+        nCol = y.shape[1]
+        print(f"y was corrected, please confirm that {nRow} row, {nCol} col")
+    else:
+        x = np.arange(1, y.shape[1] + 1)
+
+    if x.shape[0] != 1:
+        x = x.T
+    yMean = np.nanmean(y, axis=0)
+    if smth > 1:
+        yMean = savgol_filter(np.nanmean(y, axis=0), smth, 1)
+    else:
+        yMean = np.nanmean(y, axis=0)
+    if paraStdSem == "sem":
+        if smth > 1:
+            wings = savgol_filter(np.nanstd(y, axis=0) / np.sqrt(y.shape[0]), smth, 1)
+        else:
+            wings = np.nanstd(y, axis=0) / np.sqrt(y.shape[0])
+    elif paraStdSem == "std":
+        if smth > 1:
+            wings = savgol_filter(np.nanstd(y, axis=0), smth, 1)
+        else:
+            wings = np.nanstd(y, axis=0)
+
+    fill_kws = kwargs.get('fill_kws', {})
+    line_kws = kwargs.get('line_kws', {})
+    fill = ax.fill_between(x, yMean + wings, yMean - wings, color=acolor, alpha=alpha, lw=0,**fill_kws)
+    if line_kws != {} and not any(key.lower() in ['lw', 'linewidth'] for key in line_kws.keys()):
+        line = ax.plot(x, yMean, color=acolor, lw=1.5, ls=plotStyle, marker=plotMarker, **line_kws)
+    else:
+        line = ax.plot(x, yMean, color=acolor, ls=plotStyle, marker=plotMarker, **line_kws)
+    return line[0], fill
+# =============================================================================
+# # for plot figures {Qiu et al.2023}
+# =============================================================================
+# =============================================================================
+# plt.rcParams.update({'figure.max_open_warning': 0})
+# # Output matplotlib figure to SVG with text as text, not curves
+# plt.rcParams['svg.fonttype'] = 'none'
+# plt.rcParams['pdf.fonttype'] = 42
+#
+# plt.rc('text', usetex=False)
+# # plt.style.use('ggplot')
+# plt.style.use('science')
+# plt.rc('font', family='serif')
+# plt.rcParams.update({
+#     "font.family": "serif",   # specify font family here
+#     "font.serif": ["Arial"],  # specify font here
+#     "font.size": 11})
+# # plt.tight_layout()
+# =============================================================================
+# =============================================================================
+# # axis spine
+# # use it like: adjust_spines(ax, ['left', 'bottom'])
+# =============================================================================

{py2ls-0.1.4.9.dist-info → py2ls-0.1.5.0.dist-info}/METADATA

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

{py2ls-0.1.4.9.dist-info → py2ls-0.1.5.0.dist-info}/RECORD

@@ -108,7 +108,7 @@ py2ls/.gitattributes,sha256=Gh2-F2vCM7SZ01pX23UT8pQcmauXWfF3gwyRSb6ZAFs,66
 py2ls/.gitignore,sha256=y7GvbD_zZkjPVVIue8AyiuFkDMuUbvMaV65Lgu89To8,2763
 py2ls/LICENSE,sha256=UOZ1F5fFDe3XXvG4oNnkL1-Ecun7zpHzRxjp-XsMeAo,11324
 py2ls/README.md,sha256=CwvJWAnSXnCnrVHlnEbrxxi6MbjbE_MT6DH2D53S818,11572
-py2ls/__init__.py,sha256=47TrrzqhyLX7Y8xNnCfjTWQPlhlSsLwHDUT5OkEQ9Z0,239
+py2ls/__init__.py,sha256=R5OBEeK0uOULvUsY1YjIj-EAQKlVPW32uh_hQHuN5Bg,109
 py2ls/brain_atlas.py,sha256=w1o5EelRjq89zuFJUNSz4Da8HnTCwAwDAZ4NU4a-bAY,5486
 py2ls/correlators.py,sha256=RbOaJIPLCHJtUm5SFi_4dCJ7VFUPWR0PErfK3K26ad4,18243
 py2ls/data/.DS_Store,sha256=1lFlJ5EFymdzGAUAaI30vcaaLHt3F1LwpG7xILf9jsM,6148
@@ -116,12 +116,14 @@ py2ls/data/db2ls_sql_chtsht.json,sha256=ls9d7Sm8TLeujanWHfHlWhU85Qz1KnAizO_9X3wU
 py2ls/data/lang_code_iso639.json,sha256=qZiU7H2RLJjDMXK22C-jhwzLJCI5vKmampjB1ys4ek4,2157
 py2ls/db2ls.py,sha256=MMfFX47aIPIyu7fU9aPvX9lbPRPYOpJ_VXwlnWk-8qo,13615
 py2ls/freqanalysis.py,sha256=F4218VSPbgL5tnngh6xNCYuNnfR-F_QjECUUxrPYZss,32594
-py2ls/ips.py,sha256=I8pW4SJ11Y4okThmJmvWDdFmMa5XZR7MOhUWkqsEscA,140502
+py2ls/ips.py,sha256=6E_GBIyzMeWbdP3XoEQtyP3_IilDIFOylaxDuu4m3-4,95266
 py2ls/netfinder.py,sha256=ZsLWGYMeRuGvxj2nqE0Z8ANoaVl18Necfw0HQfh2q7I,45548
 py2ls/setuptools-70.1.0-py3-none-any.whl,sha256=2bi3cUVal8ip86s0SOvgspteEF8SKLukECi-EWmFomc,882588
 py2ls/sleep_events_detectors.py,sha256=36MCuRrpurn0Uvzpo3p3b3_JlVsRNHSWCXbJxCGM3mg,51546
+py2ls/stats.py,sha256=Wd9yCKQ_61QD29WMEgMuEcreFxF91NmlPW65iWT2B5w,39041
+py2ls/stdshade.py,sha256=lK9H98oEVLV14cYfB4BHYT5ptAI4Oia87MF86FqbPOg,6987
 py2ls/translator.py,sha256=6S7MmTZmjj8NljVmj0W5uEauu4ePxso3AMf2LvGVRQA,30516
 py2ls/wb_detector.py,sha256=7y6TmBUj9exCZeIgBAJ_9hwuhkDh1x_-yg4dvNY1_GQ,6284
-py2ls-0.1.4.9.dist-info/METADATA,sha256=TaiklHUvW4uVWJ2_hyDwiZL76ALa-7pBW5h2fE5koTg,17943
-py2ls-0.1.4.9.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-py2ls-0.1.4.9.dist-info/RECORD,,
+py2ls-0.1.5.0.dist-info/METADATA,sha256=oM0BHs6W_KAa52S4XXVGvrIxE3JBWqv5rMzNGtWhhzs,17943
+py2ls-0.1.5.0.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+py2ls-0.1.5.0.dist-info/RECORD,,