newsworthycharts 1.60.0__py3-none-any.whl → 1.61.0__py3-none-any.whl

newsworthycharts/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "1.60.0"
+__version__ = "1.61.0"
 
 from .chart import Chart
 from .choroplethmap import ChoroplethMap

newsworthycharts/choroplethmap.py

@@ -6,6 +6,7 @@ import geopandas as gpd
 import numpy as np
 import pandas as pd
 import mapclassify
+import matplotlib as mpl
 import matplotlib.patches as mpatches
 
 
@@ -50,11 +51,33 @@ class ChoroplethMap(Map):
         }
 
         patches = []  # for legend
-        if not self.categorical:
+        if not self.categorical and self.use_bins is False:
+            # We can not provide vmin/vmax to geopandas, so we need to
+            # normalize the data ourselves, otherwise the inset maps will be off
+            norm = mpl.colors.Normalize(vmin=df["data"].min(), vmax=df["data"].max())
+            mapper = mpl.cm.ScalarMappable(norm=norm, cmap=self.color_ramp)
+            df["color"] = df["data"].apply(lambda x: mapper.to_rgba(x) if not np.isnan(x) else "gainsboro")
+            df["color"] = df["color"].fillna("gainsboro")
+            args["color"] = df["color"]
+
+            _min = df["data"].min()
+            _max = df["data"].max()
+            _median = df["data"].median()
+            fmt = self._get_value_axis_formatter()
+            patches.append(mpatches.Patch(color=mapper.to_rgba(_min), label=fmt(_min)))
+            if _median != _min and _median != _max:
+                patches.append(mpatches.Patch(color=mapper.to_rgba(_median), label=fmt(_median)))
+            if _max != _min:
+                patches.append(mpatches.Patch(color=mapper.to_rgba(_max), label=fmt(_max)))
+
+        elif not self.categorical:
             # mapclassify doesn't work well with nan values,
             # but we to keep them for plotting, hence
             # this hack with cutting out nan's and re-pasting them below
             _has_value = df[~df["data"].isna()].copy()
+            if not pd.to_numeric(_has_value['data'], errors='coerce').notnull().all():
+                # We can't bin non-numeric data
+                raise ValueError("Data must be numeric")
             binning = mapclassify.classify(
                 np.asarray(_has_value["data"]),  # .astype("category")
                 self.binning_method,
@@ -75,10 +98,8 @@ class ChoroplethMap(Map):
             # args["cmap"] = self.color_ramp
             # We can not provide vmin/vmax to geopandas, so we need to
             # normalize the data ourselves, otherwise the inset maps will be off
-            import matplotlib.cm as cm
-            import matplotlib as mpl
             norm = mpl.colors.Normalize(vmin=df["data"].min(), vmax=df["data"].max())
-            mapper = cm.ScalarMappable(norm=norm, cmap=self.color_ramp)
+            mapper = mpl.cm.ScalarMappable(norm=norm, cmap=self.color_ramp)
             df["color"] = df["data"].apply(lambda x: mapper.to_rgba(x) if not np.isnan(x) else "gainsboro")
             df["color"] = df["color"].fillna("gainsboro")
             args["color"] = df["color"]
@@ -130,6 +151,7 @@ class ChoroplethMap(Map):
                     color = self._nwc_style[f"{color}_color"]
                 patch = mpatches.Patch(color=color, label=label)
                 patches.append(patch)
+        patches = list(reversed(patches))
         if self.missing_label:
             patches.append(mpatches.Patch(color="gainsboro", label=self.missing_label))
 
@@ -171,7 +193,8 @@ class ChoroplethMap(Map):
             for _line in [a, b, c, d]:
                 _line.set_visible(False)
 
-        self.ax.legend(
-            handles=patches,
-            **label_kwargs,
-        )
+        if len(patches):
+            self.ax.legend(
+                handles=patches,
+                **label_kwargs,
+            )

newsworthycharts/map.py

@@ -4,7 +4,6 @@ Base class for maps
 from .chart import Chart
 from .lib.geography import haversine
 from .translations.regions import NW_MUNI_TO_CLDR
-from fiona.errors import DriverError
 import geopandas as gpd
 import pathlib
 
@@ -88,6 +87,7 @@ class Map(Chart):
     def __init__(self, *args, **kwargs):
         super(Map, self).__init__(*args, **kwargs)
         self.bins = kwargs.get("bins", 5)
+        self.use_bins = kwargs.get("use_bins", True)
         self.binning_method = kwargs.get("binning_method", "natural_breaks")
         self.colors = kwargs.get("colors", None)
         self.color_ramp = kwargs.get("color_ramp", "YlGn")  # YlOrRd
@@ -139,12 +139,7 @@ class Map(Chart):
 
         if self.df is None:
             __dir = pathlib.Path(__file__).parent.resolve()
-            try:
-                self.df = gpd.read_file(f"{__dir}/maps/{base_map}-{subdivisions}.gpkg")
-            except DriverError:
-                raise ValueError(
-                    f"No such basemap: {_bm} (parsed as base: {base_map}, subdivisions: {subdivisions})"
-                )
+            self.df = gpd.read_file(f"{__dir}/maps/{base_map}-{subdivisions}.gpkg", engine="pyogrio")
         return base_map, subdivisions, subset, *opts
 
     def _prepare_map_data(self):

{newsworthycharts-1.60.0.dist-info → newsworthycharts-1.61.0.dist-info}/METADATA

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: newsworthycharts
-Version: 1.60.0
+Version: 1.61.0
 Summary: Matplotlib wrapper to create charts and publish them on Amazon S3
 Home-page: https://github.com/jplusplus/newsworthycharts
-Download-URL: https://github.com/jplusplus/newsworthycharts/archive/1.60.0.tar.gz
+Download-URL: https://github.com/jplusplus/newsworthycharts/archive/1.61.0.tar.gz
 Author: Jens Finnäs and Leo Wallentin, J++ Stockholm
 Author-email: stockholm@jplusplus.org
 License: MIT
@@ -21,9 +21,9 @@ Requires-Dist: python-dateutil <3,>=2
 Requires-Dist: Pillow ==10.4.0
 Requires-Dist: requests >=2.22
 Requires-Dist: matplotlib-label-lines ==0.5.1
-Requires-Dist: geopandas >1
+Requires-Dist: geopandas >=1
 Requires-Dist: mapclassify ==2.6.1
-Requires-Dist: fiona >=1.0
+Requires-Dist: puremagic >=1
 
 This  module contains methods for producing graphs and publishing them on Amazon S3, or in the location of your choice.
 
@@ -188,13 +188,14 @@ These settings are available for all chart types:
 
 **Map**
 
-- bins = 9  # Number of bins for continuous data
-- binning_method = "natural_breaks"
-- colors = None
-- color_ramp = "YlOrRd"
-- categorical = False  # If True, the map will be colored by category. If False, it will be colored by a continuous value
-- base_map = None
-- missing_label = None  # Add a label for no data
+- `use_bins` = True  # If False, the map will be colored by a continuous value
+- `bins` = 5  # Number of bins for continuous data
+- `binning_method` = "natural_breaks"
+- `colors` = None
+- `color_ramp` = "YlOrRd"
+- `categorical` = False  # If True, the map will be colored by category. If False, it will be colored by a continuous value
+- `base_map` = None
+- `missing_label` = None  # Add a label for no data
 
 `basemap` can be `{ISO}-{level}` or `{ISO}|{subset}-{level}`. 
 For example, `se-4` will show Swedish counties, while `se|03-7` will show municipalities (`se-7`) starting with `03`.
@@ -203,6 +204,10 @@ For example, `se-4` will show Swedish counties, while `se|03-7` will show munici
 
 _ Inherits from Map _
 
+**BubbleMap**
+
+_ Inherits from Map _
+
 **ProgressChart**
 
 **RangePlot**
@@ -247,6 +252,16 @@ Roadmap
 Changelog
 ---------
 
+- 1.61.0
+
+  - Added non-binned color ramp support to ChoroplethMap, with `use_bins=False`
+  - Improved data validation in ChoroplethMap
+  - Put largest value on top in choropleth map legends
+
+- 1.60.1
+
+  - Don't require fiona. Geopandas now support Pyogrio
+
 - 1.60.0
 
   - [BREAKING] The default number of bins in maps is now 5, not 9

{newsworthycharts-1.60.0.dist-info → newsworthycharts-1.61.0.dist-info}/RECORD

@@ -1,10 +1,10 @@
-newsworthycharts/__init__.py,sha256=xHFzdcFCUfaPTmQ4A76S5gg13omTVjATWqkExpWYdk8,1160
+newsworthycharts/__init__.py,sha256=jnjdyPdjfQOT6rFgjHDof_FP6br3JVLWzdGN66o7x3c,1160
 newsworthycharts/bubblemap.py,sha256=nkocWmpiFgfjEuJGAsthjY5X7Q56jXWsZHUGXw4PwgE,2587
 newsworthycharts/categoricalchart.py,sha256=LwOZ3VbNy9vzvoK0s77AkbfMt4CXVDSAhnsnBInUIrE,14764
 newsworthycharts/chart.py,sha256=VXEJe_g_KahyiUXqXnM3vDS7SQoW1Ppktcj13-A2_oc,30631
-newsworthycharts/choroplethmap.py,sha256=dVRShxjkBKa57r_Kp6ysdNCXGwXh3eZK28MvVtLe7BQ,6677
+newsworthycharts/choroplethmap.py,sha256=bCLf4kcchp1C2djg5AxcOM8BdbaMj0xg7UHrZsDafhI,8013
 newsworthycharts/datawrapper.py,sha256=RRkAVTpfP4updKxUIBaSmKuBi2RUVPaBRF8HDQhlGGA,11250
-newsworthycharts/map.py,sha256=c3VwhJFwmmWJyXJ7HvYdvJsWMTCKD9Eg5VnK6UmdDFc,6233
+newsworthycharts/map.py,sha256=c409jEO4L8Yr780sJRC0RchR44roAlgOUDAkuk1SfRg,6057
 newsworthycharts/rangeplot.py,sha256=NE1W9TnmlpK6T3RvBJOU3nd73EXqkj17OY9i5zlw_cQ,8366
 newsworthycharts/scatterplot.py,sha256=6iaMoiZx__Gc-2Hcdw-8Ga5dSonrFo3oexKNmSFuir4,4959
 newsworthycharts/seasonalchart.py,sha256=rr55yqJUkaYDR9Ik98jes6574oY1U8t8LwoLE3gClW4,1967
@@ -28,8 +28,8 @@ newsworthycharts/rc/newsworthy,sha256=X0btLNrmk2DRrfOsKj_WCSIgeD6btacEN2tRF_B4m8
 newsworthycharts/translations/datawrapper_regions.csv,sha256=fzZcQRX6RFMlNNP8mpgfYNdR3Y0QAlQxDXk8FXTaWWI,9214
 newsworthycharts/translations/regions.py,sha256=Nv1McQjggD4S3JRu82rDMTG3pqUVR13E5-FBpSYbm98,239
 newsworthycharts/translations/se_municipalities.csv,sha256=br_mm-IvzQtj_W55_ATREhJ97jWnCweBFlDAVY2EBxA,7098
-newsworthycharts-1.60.0.dist-info/LICENSE.txt,sha256=Sq6kGICrehbhC_FolNdXf0djKjTpv3YqjFCIYsxdQN4,1069
-newsworthycharts-1.60.0.dist-info/METADATA,sha256=_muVSibMISGRnrB1Mm2AakHaLJoPtwgQlOKjIAsHZdI,28168
-newsworthycharts-1.60.0.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
-newsworthycharts-1.60.0.dist-info/top_level.txt,sha256=dn_kzIj8UgUCMsh1PHdVEQJHVGSsN7Z8YJF-8xXa8n0,17
-newsworthycharts-1.60.0.dist-info/RECORD,,
+newsworthycharts-1.61.0.dist-info/LICENSE.txt,sha256=Sq6kGICrehbhC_FolNdXf0djKjTpv3YqjFCIYsxdQN4,1069
+newsworthycharts-1.61.0.dist-info/METADATA,sha256=gyom0iFYdCZuYaLwa24Arkma11A1Qn5H0DdrZhVJkz0,28560
+newsworthycharts-1.61.0.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
+newsworthycharts-1.61.0.dist-info/top_level.txt,sha256=dn_kzIj8UgUCMsh1PHdVEQJHVGSsN7Z8YJF-8xXa8n0,17
+newsworthycharts-1.61.0.dist-info/RECORD,,