iqplot 0.3.6__tar.gz → 0.3.7__tar.gz

{iqplot-0.3.6/iqplot.egg-info → iqplot-0.3.7}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: iqplot
-Version: 0.3.6
+Version: 0.3.7
 Summary: Generate Bokeh plots for data sets with one quantitative variable.
 Home-page: https://github.com/justinbois/iqplot
-Download-URL: https://github.com/justinbois/iqplot/tarball/0.3.6
+Download-URL: https://github.com/justinbois/iqplot/tarball/0.3.7
 Author: Justin Bois
 Author-email: bois@caltech.edu
 License: BSD

{iqplot-0.3.6 → iqplot-0.3.7}/iqplot/__init__.py

@@ -7,4 +7,4 @@ from .dist import *
 
 __author__ = """Justin Bois"""
 __email__ = "bois@caltech.edu"
-__version__ = "0.3.6"
+__version__ = "0.3.7"

{iqplot-0.3.6 → iqplot-0.3.7}/iqplot/dist.py

@@ -163,6 +163,22 @@ def ecdf(
     output : bokeh.plotting.Figure instance
         Plot populated with ECDFs.
     """
+
+    # In the future, we will have a max_points kwarg that will lead to
+    # thinning of the ECDF. The doc string for the parameter is:
+        # max_points : int or None, default None, NOT YET IMPLEMENTED
+        # Maximum number of data points to use in a single ECDF. If the
+        # number of data points exceeds max_points, then the plot is
+        # constructed where points corresponding to the quantiles
+        # [1, 2, ..., max_points] / max_points are plotted. The upper
+        # and lower extreme points are always included. If this thinning
+        # is applied, then the resulting plotted ECDF will always lie
+        # slightly below the non-thinned ECDF. If None and the number of
+        # points for an ECDF is greater than 2000, there will be a
+        # warning that the number of points exceeds the maximum. The
+        # warning will not be issued if max_points an `int`.
+
+
     # Protect against mutability of dicts
     marker_kwargs = copy.copy(marker_kwargs)
     line_kwargs = copy.copy(line_kwargs)
@@ -1149,7 +1165,11 @@ def spike(
         else:
             ptiles = np.sort(ptiles)
 
-    one_cat = cats is None or len(data.groupby(cats)) == 1
+    # Quick check to see if there is only one category.
+    # This is not an ideal way to do it, since we convert to Pandas df
+    # just for the check, which we do later as well
+    data_check, _, _, _ = utils._data_cats(data, q, cats, False, None)
+    one_cat = cats is None or len(data_check.groupby(cats)) == 1
 
     if arrangement is None:
         arrangement = "overlay" if one_cat else "stack"

{iqplot-0.3.6 → iqplot-0.3.7}/iqplot/utils.py

@@ -4,7 +4,6 @@ import warnings
 
 import numpy as np
 import pandas as pd
-import xarray
 
 import bokeh.core.enums
 
@@ -90,7 +89,7 @@ def _parse_deprecations(
 
 
 def _data_cats(data, q, cats, show_legend, legend_label):
-    if type(data) == xarray.core.dataarray.DataArray:
+    if 'xarray.core.dataarray.DataArray' in str(type(data)):
         if q is None:
             if data.name is None:
                 q = "x"
@@ -103,6 +102,25 @@ def _data_cats(data, q, cats, show_legend, legend_label):
         data = pd.DataFrame({q: data.squeeze()})
         if cats is not None:
             raise RuntimeError("If `data` is a Numpy array, `cats` must be None.")
+    elif 'polars.dataframe.frame.DataFrame' in str(type(data)):
+        # For now, we just convert to Pandas. We will add functionality
+        # to work with polars data frames to take advantage of their
+        # performance in the future.
+        data = data.to_pandas()
+    elif 'polars.series.series.Series' in str(type(data)) or type(data) == pd.core.series.Series:
+        # For now, just convert to Pandas series if it's a Polars series
+        if 'polars.series.series.Series' in str(type(data)):
+            data = data.to_pandas()
+        if q is None:
+            if data.name is None:
+                q = 'x'
+            else:
+                q = data.name
+        data = pd.DataFrame({q: data})
+        if cats is not None:
+            raise RuntimeError("If `data` is a Pandas series, `cats` must be None.")
+    elif type(data) != pd.core.frame.DataFrame:
+        raise RuntimeError(f"Data type {type(data)} for argument `data` is not supported.")
 
     # Make a copy of the data frame
     data = data.copy()

{iqplot-0.3.6 → iqplot-0.3.7/iqplot.egg-info}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: iqplot
-Version: 0.3.6
+Version: 0.3.7
 Summary: Generate Bokeh plots for data sets with one quantitative variable.
 Home-page: https://github.com/justinbois/iqplot
-Download-URL: https://github.com/justinbois/iqplot/tarball/0.3.6
+Download-URL: https://github.com/justinbois/iqplot/tarball/0.3.7
 Author: Justin Bois
 Author-email: bois@caltech.edu
 License: BSD

{iqplot-0.3.6 → iqplot-0.3.7}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 from codecs import open
 from os import path
 
-__version__ = '0.3.6'
+__version__ = '0.3.7'
 
 here = path.abspath(path.dirname(__file__))