Class hvPlotTabularPolars

hvplot/plotting/core.py:1582–1630

class hvPlotTabularPolars(hvPlotTabular):
    def _get_converter(self, x=None, y=None, kind=None, **kwds):
        import polars as pl

        params = dict(self._metadata, **kwds)
        x = x or params.pop('x', None)
        y = y or params.pop('y', None)
        kind = kind or params.pop('kind', None)

        # Find columns which should be converted for LazyDataFrame and DataFrame
        if isinstance(self._data, (pl.LazyFrame, pl.DataFrame)):
            try:
                column_names = self._data.collect_schema().names()
            except Exception:  # Maybe not always supported, has been there since 1.7.1
                column_names = list(self._data.columns)

            if params.get('hover_cols') == 'all':
                columns = column_names
            else:
                possible_columns = [
                    [v] if isinstance(v, str) else v
                    for v in params.values()
                    if isinstance(v, (str, list))
                ]

                columns = set(column_names) & set(itertools.chain(*possible_columns))
                columns = columns or {column_names[0]}
                if y is None:
                    # When y is not specified HoloViewsConverter finds all the numeric
                    # columns and use them as y values (see _process_chart_y). We need
                    # to include these columns too.
                    columns |= set(self._data.select(pl.col(pl.NUMERIC_DTYPES)).columns)
                xs = x if is_list_like(x) else (x,)
                ys = y if is_list_like(y) else (y,)
                columns |= {*xs, *ys}
                columns.discard(None)
                # Reorder the columns as in the data.
                columns = sorted(columns, key=lambda c: column_names.index(c))

        if isinstance(self._data, pl.DataFrame):
            data = self._data.select(columns).to_pandas()
        elif isinstance(self._data, pl.Series):
            data = self._data.to_pandas()
        elif isinstance(self._data, pl.LazyFrame):
            data = self._data.select(columns).collect().to_pandas()
        else:
            raise ValueError('Only Polars DataFrame, Series, and LazyFrame are supported')

        return HoloViewsConverter(data, x, y, kind=kind, **params)


class hvPlot(hvPlotTabular):