hvplot 0.9.3a1__py3-none-any.whl

hvplot/plotting/__init__.py

@@ -0,0 +1,63 @@
+import holoviews as hv
+from ..util import with_hv_extension, is_polars
+
+from .core import hvPlot, hvPlotTabular  # noqa
+
+from .andrews_curves import andrews_curves  # noqa
+from .parallel_coordinates import parallel_coordinates  # noqa
+from .lag_plot import lag_plot  # noqa
+from .scatter_matrix import scatter_matrix  # noqa
+
+
+@with_hv_extension
+def plot(data, kind, **kwargs):
+    # drop reuse_plot
+    kwargs.pop('reuse_plot', None)
+
+    # replace with shared_axes
+    sharex = kwargs.pop('sharex', None)
+    sharey = kwargs.pop('sharey', None)
+    if sharex is not None and sharey is not None:
+        kwargs['shared_axes'] = sharex or sharey
+    elif sharex is not None:
+        kwargs['shared_axes'] = sharex
+    elif sharey is not None:
+        kwargs['shared_axes'] = sharey
+
+    # drop all kwargs that are set to None
+    no_none_kwargs = {}
+    for k, v in kwargs.items():
+        if v is not None:
+            no_none_kwargs[k] = v
+
+    if is_polars(data):
+        from .core import hvPlotTabularPolars
+
+        return hvPlotTabularPolars(data)(kind=kind, **no_none_kwargs)
+    return hvPlotTabular(data)(kind=kind, **no_none_kwargs)
+
+
+def boxplot_series(*args, **kwargs):
+    return plot(*args, kind='box', **kwargs)
+
+
+def boxplot_frame(*args, **kwargs):
+    return plot(*args, kind='box', **kwargs)
+
+
+def boxplot_frame_groupby(grouped, **kwargs):
+    width = kwargs.pop('width', 300)
+    subplots = kwargs.pop('subplots', True)
+    layout = hv.Layout if subplots else hv.Overlay
+    plots = [
+        plot(data=data, kind='box', title=name, width=width, **kwargs) for name, data in grouped
+    ]
+    return layout(plots)
+
+
+def hist_series(*args, **kwargs):
+    return plot(*args, kind='hist', **kwargs)
+
+
+def hist_frame(*args, **kwargs):
+    return plot(*args, kind='hist', **kwargs)

hvplot/plotting/andrews_curves.py

@@ -0,0 +1,99 @@
+import numpy as np
+import pandas as pd
+
+import holoviews as hv
+import colorcet as cc
+
+from ..backend_transforms import _transfer_opts_cur_backend
+from ..util import with_hv_extension
+
+
+@with_hv_extension
+def andrews_curves(
+    data,
+    class_column,
+    samples=200,
+    alpha=0.5,
+    width=600,
+    height=300,
+    cmap=None,
+    colormap=None,
+    **kwds,
+):
+    """
+    Generate a plot of Andrews curves, for visualising clusters of
+    multivariate data.
+
+    Andrews curves have the functional form:
+
+    f(t) = x_1/sqrt(2) + x_2 sin(t) + x_3 cos(t) +
+           x_4 sin(2t) + x_5 cos(2t) + ...
+
+    Where x coefficients correspond to the values of each dimension and t is
+    linearly spaced between -pi and +pi. Each row of frame then corresponds to
+    a single curve.
+
+    Parameters
+    ----------
+    frame: DataFrame
+        Data to be plotted, preferably normalized to (0.0, 1.0)
+    class_column: str
+        Column name containing class names
+    samples: int, optional
+        Number of samples to draw
+    alpha: float, optional
+        The transparency of the lines
+    cmap/colormap: str or colormap object
+        Colormap to use for groups
+
+    Returns
+    -------
+    obj : HoloViews object
+        The HoloViews representation of the plot.
+
+    See Also
+    --------
+    pandas.plotting.parallel_coordinates : matplotlib version of this routine
+    """
+    t = np.linspace(-np.pi, np.pi, samples)
+    vals = data.drop(class_column, axis=1).values.T
+
+    curves = np.outer(vals[0], np.ones_like(t))
+    for i in range(1, len(vals)):
+        ft = ((i + 1) // 2) * t
+        if i % 2 == 1:
+            curves += np.outer(vals[i], np.sin(ft))
+        else:
+            curves += np.outer(vals[i], np.cos(ft))
+
+    df = pd.DataFrame(
+        {
+            't': np.tile(np.arange(samples), curves.shape[0]),
+            'sample': np.repeat(np.arange(curves.shape[0]), curves.shape[1]),
+            'value': curves.ravel(),
+            class_column: np.repeat(data[class_column], samples),
+        }
+    )
+
+    labelled = ['x']
+    options = {
+        'Overlay': dict(legend_limit=5000),
+        'Curve': dict(kwds, labelled=labelled, alpha=alpha, width=width, height=height, **kwds),
+    }
+    dataset = hv.Dataset(df)
+    groups = dataset.to(hv.Curve, 't', 'value').overlay('sample').items()
+
+    if cmap and colormap:
+        raise TypeError('Only specify one of `cmap` and `colormap`.')
+    cmap = cmap or colormap or cc.palette['glasbey_category10']
+    colors = hv.plotting.util.process_cmap(cmap, categorical=True, ncolors=len(groups))
+
+    el = hv.Overlay(
+        [
+            curve.relabel(k).options('Curve', color=c, backend='bokeh')
+            for c, (k, v) in zip(colors, groups)
+            for curve in v
+        ]
+    ).options(options, backend='bokeh')
+    el = _transfer_opts_cur_backend(el)
+    return el