figrecipe 0.5.0__py3-none-any.whl → 0.7.4__py3-none-any.whl

figrecipe/_dev/demo_plotters/image_matrix/plot_imshow.py

@@ -0,0 +1,23 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""imshow: image display demo."""
+
+
+def plot_imshow(plt, rng, ax=None):
+    """Image display demo.
+
+    Demonstrates: ax.imshow()
+    """
+    if ax is None:
+        fig, ax = plt.subplots()
+    else:
+        fig = ax.get_figure() if hasattr(ax, "get_figure") else ax.fig
+
+    data = rng.uniform(0, 1, (20, 20))
+    ax.imshow(data, id="imshow")
+    ax.axis("off")
+    ax.set_title("imshow")
+    return fig, ax
+
+
+# EOF

figrecipe/_dev/demo_plotters/image_matrix/plot_matshow.py

@@ -0,0 +1,23 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""matshow: matrix display demo."""
+
+
+def plot_matshow(plt, rng, ax=None):
+    """Matrix display demo.
+
+    Demonstrates: ax.matshow()
+    """
+    if ax is None:
+        fig, ax = plt.subplots()
+    else:
+        fig = ax.get_figure() if hasattr(ax, "get_figure") else ax.fig
+
+    data = rng.uniform(0, 1, (10, 10))
+    ax.matshow(data, id="matshow")
+    ax.axis("off")
+    ax.set_title("matshow")
+    return fig, ax
+
+
+# EOF

figrecipe/_dev/demo_plotters/image_matrix/plot_pcolor.py

@@ -0,0 +1,29 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""pcolor: pseudocolor plot demo."""
+
+import numpy as np
+
+
+def plot_pcolor(plt, rng, ax=None):
+    """Pseudocolor plot demo.
+
+    Demonstrates: ax.pcolor()
+    """
+    if ax is None:
+        fig, ax = plt.subplots()
+    else:
+        fig = ax.get_figure() if hasattr(ax, "get_figure") else ax.fig
+
+    x = np.arange(0, 10, 1)
+    y = np.arange(0, 10, 1)
+    X, Y = np.meshgrid(x, y)
+    Z = np.sin(X) * np.cos(Y)
+    ax.pcolor(X, Y, Z, id="pcolor")
+    ax.set_xlabel("X")
+    ax.set_ylabel("Y")
+    ax.set_title("pcolor")
+    return fig, ax
+
+
+# EOF

figrecipe/_dev/demo_plotters/image_matrix/plot_pcolormesh.py

@@ -0,0 +1,29 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""pcolormesh: pseudocolor mesh plot demo."""
+
+import numpy as np
+
+
+def plot_pcolormesh(plt, rng, ax=None):
+    """Pseudocolor mesh plot demo.
+
+    Demonstrates: ax.pcolormesh()
+    """
+    if ax is None:
+        fig, ax = plt.subplots()
+    else:
+        fig = ax.get_figure() if hasattr(ax, "get_figure") else ax.fig
+
+    x = np.linspace(0, 2 * np.pi, 50)
+    y = np.linspace(0, 2 * np.pi, 50)
+    X, Y = np.meshgrid(x, y)
+    Z = np.sin(X) * np.cos(Y)
+    ax.pcolormesh(X, Y, Z, id="pcolormesh")
+    ax.set_xlabel("X")
+    ax.set_ylabel("Y")
+    ax.set_title("pcolormesh")
+    return fig, ax
+
+
+# EOF

figrecipe/_dev/demo_plotters/image_matrix/plot_spy.py

@@ -0,0 +1,29 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""spy: sparse matrix visualization demo."""
+
+import numpy as np
+
+
+def plot_spy(plt, rng, ax=None):
+    """Sparse matrix visualization demo.
+
+    Demonstrates: ax.spy()
+    """
+    if ax is None:
+        fig, ax = plt.subplots()
+    else:
+        fig = ax.get_figure() if hasattr(ax, "get_figure") else ax.fig
+
+    # Create sparse-like matrix
+    data = np.zeros((20, 20))
+    for _ in range(50):
+        i, j = rng.integers(0, 20, 2)
+        data[i, j] = 1
+    ax.spy(data, id="spy")
+    ax.axis("off")
+    ax.set_title("spy")
+    return fig, ax
+
+
+# EOF

figrecipe/_dev/demo_plotters/line_curve/__init__.py

@@ -0,0 +1,4 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""Demo plotters for line_curve category."""
+

figrecipe/_dev/demo_plotters/line_curve/plot_errorbar.py

@@ -0,0 +1,28 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""errorbar: error bar plot demo."""
+
+import numpy as np
+
+
+def plot_errorbar(plt, rng, ax=None):
+    """Error bar plot demo.
+
+    Demonstrates: ax.errorbar()
+    """
+    if ax is None:
+        fig, ax = plt.subplots()
+    else:
+        fig = ax.get_figure() if hasattr(ax, "get_figure") else ax.fig
+
+    x = np.arange(5)
+    y = rng.uniform(2, 8, 5)
+    yerr = rng.uniform(0.3, 0.8, 5)
+    ax.errorbar(x, y, yerr=yerr, id="errorbar")
+    ax.set_xlabel("X")
+    ax.set_ylabel("Y")
+    ax.set_title("errorbar")
+    return fig, ax
+
+
+# EOF

figrecipe/_dev/demo_plotters/line_curve/plot_fill.py

@@ -0,0 +1,29 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""fill: filled polygon demo."""
+
+import numpy as np
+
+
+def plot_fill(plt, rng, ax=None):
+    """Filled polygon demo.
+
+    Demonstrates: ax.fill()
+    """
+    if ax is None:
+        fig, ax = plt.subplots()
+    else:
+        fig = ax.get_figure() if hasattr(ax, "get_figure") else ax.fig
+
+    theta = np.linspace(0, 2 * np.pi, 6)
+    x = np.cos(theta)
+    y = np.sin(theta)
+    ax.fill(x, y, alpha=0.5, id="fill")
+    ax.set_xlabel("X")
+    ax.set_ylabel("Y")
+    ax.set_title("fill")
+    ax.set_aspect("equal")
+    return fig, ax
+
+
+# EOF

figrecipe/_dev/demo_plotters/line_curve/plot_fill_between.py

@@ -0,0 +1,30 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""fill_between: filled area between curves demo."""
+
+import numpy as np
+
+
+def plot_fill_between(plt, rng, ax=None):
+    """Filled area between curves demo.
+
+    Demonstrates: ax.fill_between()
+    """
+    if ax is None:
+        fig, ax = plt.subplots()
+    else:
+        fig = ax.get_figure() if hasattr(ax, "get_figure") else ax.fig
+
+    x = np.linspace(0, 10, 100)
+    y1 = np.sin(x)
+    y2 = np.sin(x) + 0.5
+    ax.fill_between(x, y1, y2, alpha=0.5, id="fill_between")
+    ax.plot(x, y1, id="lower")
+    ax.plot(x, y2, id="upper")
+    ax.set_xlabel("X")
+    ax.set_ylabel("Y")
+    ax.set_title("fill_between")
+    return fig, ax
+
+
+# EOF

figrecipe/_dev/demo_plotters/line_curve/plot_fill_betweenx.py

@@ -0,0 +1,28 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""fill_betweenx: horizontal filled area demo."""
+
+import numpy as np
+
+
+def plot_fill_betweenx(plt, rng, ax=None):
+    """Horizontal filled area demo.
+
+    Demonstrates: ax.fill_betweenx()
+    """
+    if ax is None:
+        fig, ax = plt.subplots()
+    else:
+        fig = ax.get_figure() if hasattr(ax, "get_figure") else ax.fig
+
+    y = np.linspace(0, 10, 100)
+    x1 = np.sin(y)
+    x2 = np.sin(y) + 0.5
+    ax.fill_betweenx(y, x1, x2, alpha=0.5, id="fill_betweenx")
+    ax.set_xlabel("X")
+    ax.set_ylabel("Y")
+    ax.set_title("fill_betweenx")
+    return fig, ax
+
+
+# EOF

figrecipe/_dev/demo_plotters/line_curve/plot_plot.py

@@ -0,0 +1,28 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""plot: line plot demo."""
+
+import numpy as np
+
+
+def plot_plot(plt, rng, ax=None):
+    """Line plot demo.
+
+    Demonstrates: ax.plot()
+    """
+    if ax is None:
+        fig, ax = plt.subplots()
+    else:
+        fig = ax.get_figure() if hasattr(ax, "get_figure") else ax.fig
+
+    x = np.linspace(0, 2 * np.pi, 100)
+    ax.plot(x, np.sin(x), label="sin", id="sin")
+    ax.plot(x, np.cos(x), label="cos", id="cos")
+    ax.legend()
+    ax.set_xlabel("X")
+    ax.set_ylabel("Y")
+    ax.set_title("plot")
+    return fig, ax
+
+
+# EOF

figrecipe/_dev/demo_plotters/line_curve/plot_stackplot.py

@@ -0,0 +1,29 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""stackplot: stacked area plot demo."""
+
+import numpy as np
+
+
+def plot_stackplot(plt, rng, ax=None):
+    """Stacked area plot demo.
+
+    Demonstrates: ax.stackplot()
+    """
+    if ax is None:
+        fig, ax = plt.subplots()
+    else:
+        fig = ax.get_figure() if hasattr(ax, "get_figure") else ax.fig
+
+    x = np.arange(10)
+    y1 = rng.uniform(1, 3, 10)
+    y2 = rng.uniform(1, 3, 10)
+    y3 = rng.uniform(1, 3, 10)
+    ax.stackplot(x, y1, y2, y3, id="stackplot")
+    ax.set_xlabel("X")
+    ax.set_ylabel("Y")
+    ax.set_title("stackplot")
+    return fig, ax
+
+
+# EOF

figrecipe/_dev/demo_plotters/line_curve/plot_stairs.py

@@ -0,0 +1,27 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""stairs: staircase plot demo."""
+
+import numpy as np
+
+
+def plot_stairs(plt, rng, ax=None):
+    """Staircase plot demo.
+
+    Demonstrates: ax.stairs()
+    """
+    if ax is None:
+        fig, ax = plt.subplots()
+    else:
+        fig = ax.get_figure() if hasattr(ax, "get_figure") else ax.fig
+
+    values = rng.uniform(1, 5, 10)
+    edges = np.arange(11)
+    ax.stairs(values, edges, fill=True, alpha=0.5, id="stairs")
+    ax.set_xlabel("X")
+    ax.set_ylabel("Y")
+    ax.set_title("stairs")
+    return fig, ax
+
+
+# EOF

figrecipe/_dev/demo_plotters/line_curve/plot_step.py

@@ -0,0 +1,27 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""step: step plot demo."""
+
+import numpy as np
+
+
+def plot_step(plt, rng, ax=None):
+    """Step plot demo.
+
+    Demonstrates: ax.step()
+    """
+    if ax is None:
+        fig, ax = plt.subplots()
+    else:
+        fig = ax.get_figure() if hasattr(ax, "get_figure") else ax.fig
+
+    x = np.arange(10)
+    y = rng.uniform(0, 1, 10)
+    ax.step(x, y, where="mid", id="step")
+    ax.set_xlabel("X")
+    ax.set_ylabel("Y")
+    ax.set_title("step")
+    return fig, ax
+
+
+# EOF

figrecipe/_dev/demo_plotters/scatter_points/__init__.py

@@ -0,0 +1,4 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""Demo plotters for scatter_points category."""
+

figrecipe/_dev/demo_plotters/scatter_points/plot_scatter.py

@@ -0,0 +1,24 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""scatter: scatter plot demo."""
+
+
+def plot_scatter(plt, rng, ax=None):
+    """Scatter plot demo.
+
+    Demonstrates: ax.scatter()
+    """
+    if ax is None:
+        fig, ax = plt.subplots()
+    else:
+        fig = ax.get_figure() if hasattr(ax, "get_figure") else ax.fig
+
+    x, y = rng.uniform(0, 10, 50), rng.uniform(0, 10, 50)
+    ax.scatter(x, y, id="scatter")
+    ax.set_xlabel("X")
+    ax.set_ylabel("Y")
+    ax.set_title("scatter")
+    return fig, ax
+
+
+# EOF

figrecipe/_dev/demo_plotters/special/__init__.py

@@ -0,0 +1,4 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""Demo plotters for special category."""
+

figrecipe/_dev/demo_plotters/special/plot_eventplot.py

@@ -0,0 +1,25 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""eventplot: event/spike raster demo."""
+
+
+def plot_eventplot(plt, rng, ax=None):
+    """Event/spike raster demo.
+
+    Demonstrates: ax.eventplot()
+    """
+    if ax is None:
+        fig, ax = plt.subplots()
+    else:
+        fig = ax.get_figure() if hasattr(ax, "get_figure") else ax.fig
+
+    # Multiple event sequences
+    events = [rng.uniform(0, 10, 20) for _ in range(5)]
+    ax.eventplot(events, id="eventplot")
+    ax.set_xlabel("Time")
+    ax.set_ylabel("Channel")
+    ax.set_title("eventplot")
+    return fig, ax
+
+
+# EOF

figrecipe/_dev/demo_plotters/special/plot_loglog.py

@@ -0,0 +1,27 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""loglog: log-log plot demo."""
+
+import numpy as np
+
+
+def plot_loglog(plt, rng, ax=None):
+    """Log-log plot demo.
+
+    Demonstrates: ax.loglog()
+    """
+    if ax is None:
+        fig, ax = plt.subplots()
+    else:
+        fig = ax.get_figure() if hasattr(ax, "get_figure") else ax.fig
+
+    x = np.logspace(0, 3, 50)
+    y = x**2
+    ax.loglog(x, y, id="loglog")
+    ax.set_xlabel("X")
+    ax.set_ylabel("Y")
+    ax.set_title("loglog")
+    return fig, ax
+
+
+# EOF

figrecipe/_dev/demo_plotters/special/plot_pie.py

@@ -0,0 +1,27 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""pie: pie chart demo."""
+
+import matplotlib.pyplot as mpl_plt
+
+
+def plot_pie(plt, rng, ax=None):
+    """Pie chart demo.
+
+    Demonstrates: ax.pie()
+    """
+    if ax is None:
+        fig, ax = plt.subplots()
+    else:
+        fig = ax.get_figure() if hasattr(ax, "get_figure") else ax.fig
+
+    sizes = [35, 25, 20, 20]
+    labels = ["A", "B", "C", "D"]
+    # Get colors from current style's color cycle
+    colors = mpl_plt.rcParams["axes.prop_cycle"].by_key()["color"][:4]
+    ax.pie(sizes, labels=labels, colors=colors, id="pie")
+    ax.set_title("pie")
+    return fig, ax
+
+
+# EOF

figrecipe/_dev/demo_plotters/special/plot_semilogx.py

@@ -0,0 +1,27 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""semilogx: semi-log X plot demo."""
+
+import numpy as np
+
+
+def plot_semilogx(plt, rng, ax=None):
+    """Semi-log X plot demo.
+
+    Demonstrates: ax.semilogx()
+    """
+    if ax is None:
+        fig, ax = plt.subplots()
+    else:
+        fig = ax.get_figure() if hasattr(ax, "get_figure") else ax.fig
+
+    x = np.logspace(0, 3, 50)
+    y = np.log10(x)
+    ax.semilogx(x, y, id="semilogx")
+    ax.set_xlabel("X (log)")
+    ax.set_ylabel("Y")
+    ax.set_title("semilogx")
+    return fig, ax
+
+
+# EOF

figrecipe/_dev/demo_plotters/special/plot_semilogy.py

@@ -0,0 +1,27 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""semilogy: semi-log Y plot demo."""
+
+import numpy as np
+
+
+def plot_semilogy(plt, rng, ax=None):
+    """Semi-log Y plot demo.
+
+    Demonstrates: ax.semilogy()
+    """
+    if ax is None:
+        fig, ax = plt.subplots()
+    else:
+        fig = ax.get_figure() if hasattr(ax, "get_figure") else ax.fig
+
+    x = np.linspace(0, 5, 50)
+    y = np.exp(x)
+    ax.semilogy(x, y, id="semilogy")
+    ax.set_xlabel("X")
+    ax.set_ylabel("Y (log)")
+    ax.set_title("semilogy")
+    return fig, ax
+
+
+# EOF

figrecipe/_dev/demo_plotters/special/plot_stem.py

@@ -0,0 +1,27 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""stem: stem plot demo."""
+
+import numpy as np
+
+
+def plot_stem(plt, rng, ax=None):
+    """Stem plot demo.
+
+    Demonstrates: ax.stem()
+    """
+    if ax is None:
+        fig, ax = plt.subplots()
+    else:
+        fig = ax.get_figure() if hasattr(ax, "get_figure") else ax.fig
+
+    x = np.arange(10)
+    y = rng.uniform(0, 1, 10)
+    ax.stem(x, y, id="stem")
+    ax.set_xlabel("X")
+    ax.set_ylabel("Y")
+    ax.set_title("stem")
+    return fig, ax
+
+
+# EOF

figrecipe/_dev/demo_plotters/spectral_signal/__init__.py

@@ -0,0 +1,4 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""Demo plotters for spectral_signal category."""
+

figrecipe/_dev/demo_plotters/spectral_signal/plot_acorr.py

@@ -0,0 +1,24 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""acorr: autocorrelation demo."""
+
+
+def plot_acorr(plt, rng, ax=None):
+    """Autocorrelation demo.
+
+    Demonstrates: ax.acorr()
+    """
+    if ax is None:
+        fig, ax = plt.subplots()
+    else:
+        fig = ax.get_figure() if hasattr(ax, "get_figure") else ax.fig
+
+    x = rng.normal(0, 1, 100)
+    ax.acorr(x, maxlags=50, id="acorr")
+    ax.set_xlabel("Lag")
+    ax.set_ylabel("Autocorrelation")
+    ax.set_title("acorr")
+    return fig, ax
+
+
+# EOF

figrecipe/_dev/demo_plotters/spectral_signal/plot_angle_spectrum.py

@@ -0,0 +1,28 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""angle_spectrum: phase angle spectrum demo."""
+
+import numpy as np
+
+
+def plot_angle_spectrum(plt, rng, ax=None):
+    """Phase angle spectrum demo.
+
+    Demonstrates: ax.angle_spectrum()
+    """
+    if ax is None:
+        fig, ax = plt.subplots()
+    else:
+        fig = ax.get_figure() if hasattr(ax, "get_figure") else ax.fig
+
+    fs = 1000
+    t = np.linspace(0, 1, fs)
+    signal = np.sin(2 * np.pi * 50 * t) + 0.5 * np.sin(2 * np.pi * 120 * t + np.pi / 3)
+    ax.angle_spectrum(signal, Fs=fs, id="angle_spectrum")
+    ax.set_xlabel("Frequency")
+    ax.set_ylabel("Phase (radians)")
+    ax.set_title("angle_spectrum")
+    return fig, ax
+
+
+# EOF

figrecipe/_dev/demo_plotters/spectral_signal/plot_cohere.py

@@ -0,0 +1,29 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""cohere: coherence demo."""
+
+import numpy as np
+
+
+def plot_cohere(plt, rng, ax=None):
+    """Coherence demo.
+
+    Demonstrates: ax.cohere()
+    """
+    if ax is None:
+        fig, ax = plt.subplots()
+    else:
+        fig = ax.get_figure() if hasattr(ax, "get_figure") else ax.fig
+
+    fs = 1000
+    t = np.linspace(0, 1, fs)
+    x = np.sin(2 * np.pi * 50 * t) + rng.normal(0, 0.3, len(t))
+    y = np.sin(2 * np.pi * 50 * t + np.pi / 4) + rng.normal(0, 0.3, len(t))
+    ax.cohere(x, y, Fs=fs, id="cohere")
+    ax.set_xlabel("Frequency")
+    ax.set_ylabel("Coherence")
+    ax.set_title("cohere")
+    return fig, ax
+
+
+# EOF