newsworthycharts 1.65.4__py3-none-any.whl → 1.67.0__py3-none-any.whl

newsworthycharts/__init__.py

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

newsworthycharts/categoricalchart.py

@@ -1,6 +1,6 @@
 from .chart import Chart
 from .lib.utils import to_float
-
+from adjustText import adjust_text
 import numpy as np
 
 
@@ -16,6 +16,7 @@ class CategoricalChart(Chart):
         self.stacked = False
         self.legend = True
         self.type = "bars"
+        self.label_placement = None  # legend|inline|outside  # defaults will vary
 
         # Optional: specify a list of colors (for multiple datasets)
         self.colors = None
@@ -25,32 +26,49 @@ class CategoricalChart(Chart):
         return self.stacked
 
     def _add_pie_data(self):
-        if len(self.data) > 1:
-            raise ValueError("Pie chart takes one data series only.")
-        self.legend = False
         self.show_ticks = False
 
-        data = self.data[0]
-        values = [to_float(x[1]) for x in data]
-        labels = [x[0] for x in data]
-
-        if self.colors is not None:
-            colors = self.colors
-        else:
-            colors = [self._nwc_style["qualitative_colors"][i] for i in range(len(values))]
-
-        explode = [0.1 if x == self.highlight else 0 for x in labels]
+        if len(self.data) > 1:
+            subaxes = self._fig.subplots(1, len(self.data))
+        patches = []
+        for idx, serie in enumerate(self.data):
+            if len(self.data) == 1:
+                subax = self.ax
+            else:
+                subax = subaxes[idx]
 
-        self.ax.pie(
-            values,
-            labels=labels,
-            startangle=90,
-            colors=colors,
-            explode=explode
-        )
-        self.ax.axis('equal')
+            values = [to_float(x[1]) for x in serie]
+            labels = [x[0] for x in serie]
 
-        self._setup_legend()
+            if self.colors is not None:
+                colors = self.colors
+            else:
+                colors = [self._nwc_style["qualitative_colors"][i] for i in range(len(values))]
+
+            explode = [0.1 if x == self.highlight else 0 for x in labels]
+            legend_placement = self.label_placement or "inline"
+            _pie = subax.pie(
+                values,
+                labels=labels if legend_placement == "inline" else None,
+                startangle=90,
+                colors=colors,
+                explode=explode
+            )
+            if idx == 0:
+                patches += _pie[0]
+            subax.axis('equal')
+            # `labels` in pie charts are used to label the series, not the values
+            if self.labels:
+                subax.set_title(self.labels[idx], loc='center', y=0)
+        if legend_placement == "inline":
+            self.legend = False
+        elif legend_placement == "legend":
+            self.ax.legend(patches, labels, loc='best', frameon=True, fancybox=True).set_zorder(100)
+        elif legend_placement == "outside":
+            _ = self.ax.legend(patches, labels, bbox_to_anchor=(1.04, 1), loc="upper left")
+            _.set(zorder=20)
+
+        self.ax.axis('off')
 
     def _add_data(self):
         if self.type == "pie":
@@ -84,7 +102,7 @@ class CategoricalChart(Chart):
             serie_values.append(_values)
 
         cum_values = np.cumsum(serie_values, axis=0).tolist()
-
+        self._bars = []
         for i, data in enumerate(self.data):
 
             # Replace None values with 0's to be able to plot bars
@@ -160,6 +178,7 @@ class CategoricalChart(Chart):
                 bar_pos = [x + i * bar_width
                            for x in np.arange(len(values))]
 
+            _bar = None
             if self.bar_orientation == "horizontal":
                 kwargs = dict(align='center', height=bar_width,
                               color=colors, zorder=2)
@@ -167,7 +186,7 @@ class CategoricalChart(Chart):
                     # To make stacked bars we need to set bottom value
                     kwargs["left"] = cum_values[i - 1]
 
-                self.ax.barh(bar_pos, values, **kwargs)
+                _bar = self.ax.barh(bar_pos, values, **kwargs)
 
             elif self.bar_orientation == "vertical":
                 kwargs = dict(
@@ -178,7 +197,8 @@ class CategoricalChart(Chart):
                 if self.stacked and i > 0:
                     # To make stacked bars we need to set bottom value
                     kwargs["bottom"] = cum_values[i - 1]
-                self.ax.bar(bar_pos, values, **kwargs)
+                _bar = self.ax.bar(bar_pos, values, **kwargs)
+            self._bars.append(_bar)
 
         if self.bar_orientation == "horizontal":
             margin = 0.02  # above and below first/last bar on horizontal
@@ -200,9 +220,6 @@ class CategoricalChart(Chart):
             self.ax.set_xticklabels(categories)
             self.ax.xaxis.set_ticks_position('none')
 
-        if len(self.data) > 1:
-            self.ax.legend(self.labels, loc='best')
-
         self._setup_legend()
 
     def _setup_legend(self):
@@ -211,6 +228,31 @@ class CategoricalChart(Chart):
             legend = self.ax.get_legend()
             if legend:
                 legend.remove()
+        else:
+            if len(self.data) == 0 and self.type != "pie":
+                return
+            placement = self.label_placement or "legend"
+            if placement == "legend":
+                self.ax.legend(self.labels, loc='best')
+            elif placement == "outside":
+                _ = self.ax.legend(self.labels, bbox_to_anchor=(1.04, 1), loc="upper left")
+                _.set(zorder=20)
+            elif placement == "inline":
+                texts = []
+                for idx, _bar in enumerate(self._bars):
+                    _texts = self.ax.bar_label(
+                        _bar,
+                        labels=[self.labels[idx]] * self.data.num_categories,
+                        label_type="center",
+                        backgroundcolor="#f0f0f099",
+                    )
+                    if idx > 0:
+                        texts += [*_texts]
+                adjust_text(
+                    texts,
+                    ax=self.ax,
+                    autoalign="x" if self.bar_orientation == "horizontal" else "y",
+                )
 
 
 class CategoricalChartWithReference(CategoricalChart):
@@ -236,6 +278,7 @@ class CategoricalChartWithReference(CategoricalChart):
         self.value_axis.grid(True)
 
         bar_width = 0.8 / len(self.data)
+        self._bars = []
         for i, data in enumerate(self.data):
 
             # Replace None values with 0's to be able to plot bars
@@ -263,18 +306,30 @@ class CategoricalChartWithReference(CategoricalChart):
 
             zorder = len(self.data) - i
             if self.bar_orientation == "horizontal":
-                self.ax.barh(bar_pos, values, height=bar_width, align='center',
-                             color=color, zorder=zorder)
+                bar = self.ax.barh(
+                    bar_pos,
+                    values,
+                    height=bar_width,
+                    align='center',
+                    color=color,
+                    zorder=zorder,
+                )
                 self.ax.set_yticks(tick_pos)
                 self.ax.set_yticklabels(categories)
                 # self.ax.invert_yaxis()
 
             elif self.bar_orientation == "vertical":
-                self.ax.bar(bar_pos, values, width=bar_width, color=color,
-                            zorder=zorder)
+                bar = self.ax.bar(
+                    bar_pos,
+                    values,
+                    width=bar_width,
+                    color=color,
+                    zorder=zorder,
+                )
                 self.ax.set_xticks(tick_pos)
                 self.ax.set_xticklabels(categories)
                 self.ax.xaxis.set_ticks_position('none')
+            self._bars.append(bar)
 
         # Make sure labels are not cropped
         yaxis_bbox = self.ax.yaxis.get_tightbbox(self._fig.canvas.get_renderer())
@@ -282,9 +337,6 @@ class CategoricalChartWithReference(CategoricalChart):
         margin -= yaxis_bbox.min[0] / float(self._w)
         self._fig.subplots_adjust(left=margin)
 
-        if self.labels:
-            self.ax.legend(self.labels, loc='best')
-
         self._setup_legend()
 
 
@@ -339,7 +391,7 @@ class ProgressChart(CategoricalChart):
 
             # rect.set_linewidth(1)
             # rect.set_edgecolor(color_progress)
-        
+
         # LABELING: Target
         if self.target_label:
             offset = 25

newsworthycharts/chart.py

@@ -285,7 +285,7 @@ class Chart(object):
     def _add_title(self, title_text):
         """Add a title."""
         text = self._fig.suptitle(
-            title_text, wrap=True,
+            title_text.strip(), wrap=True,
             x=0,
             y=0.985,  # default: 0.98
             horizontalalignment="left",
@@ -300,7 +300,7 @@ class Chart(object):
         text = self._fig.text(
             0,
             y_pos,
-            subtitle_text,
+            subtitle_text.strip(),
             wrap=True,
             verticalalignment="top",
             linespacing=1.4,

newsworthycharts/lib/datalist.py

@@ -196,6 +196,11 @@ class DataSet(MutableSequence):
         values = array(self.values, dtype=float)
         return np.nansum(values, axis=0).tolist()
 
+    @property
+    def num_categories(self):
+        categories = set([x[0] for s in self.list for x in s])
+        return len(categories)
+
     def __len__(self):
         return len(self.list)
 

{newsworthycharts-1.65.4.dist-info → newsworthycharts-1.67.0.dist-info}/METADATA

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: newsworthycharts
-Version: 1.65.4
+Version: 1.67.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.65.4.tar.gz
+Download-URL: https://github.com/jplusplus/newsworthycharts/archive/1.67.0.tar.gz
 Author: Jens Finnäs and Leo Wallentin, J++ Stockholm
 Author-email: stockholm@jplusplus.org
 License: MIT
@@ -259,6 +259,18 @@ Roadmap
 Changelog
 ---------
 
+- 1.67.0
+
+  - Fixed label margins in pie charts
+  - Trim all titles and subtitles of leading and trailing whitespace
+  - Improved piechart annotation
+  - Support `label_placement` (legend|outside|inline) in categorical charts ('legend' is still buggy in pie charts)
+
+- 1.66.0
+
+  - Support multiple data series in pie charts (displayed as small multiples)
+
+
 - 1.65.4
 
   - Revert some of the title margin changes

{newsworthycharts-1.65.4.dist-info → newsworthycharts-1.67.0.dist-info}/RECORD

@@ -1,7 +1,7 @@
-newsworthycharts/__init__.py,sha256=dpkWjO_q656I2XsRZeAZa5WZ1kihmp5KOSxYFwCbETE,1160
+newsworthycharts/__init__.py,sha256=t7zoA4QcnDRPHIEjlFDnSD3BUJrVpJjyY2dxNvpp_MQ,1160
 newsworthycharts/bubblemap.py,sha256=nkocWmpiFgfjEuJGAsthjY5X7Q56jXWsZHUGXw4PwgE,2587
-newsworthycharts/categoricalchart.py,sha256=wLrS8H69Sx3FT-0r3w3ZOIC83IHb8HCHD0zTsXiIiNA,15737
-newsworthycharts/chart.py,sha256=FEQRAvdfX0x3KWFBxFEpGhh3w6dRMX5dTw-VIFPo4jA,34374
+newsworthycharts/categoricalchart.py,sha256=3XXBzIILkUl0721FN9j8cDK5mSjh8vK0JQGXCRu9ilo,18048
+newsworthycharts/chart.py,sha256=9M4LE5UUN5kMaVQ1jnCAKpACU3oZBMPPUpodad9sBT4,34390
 newsworthycharts/choroplethmap.py,sha256=bCLf4kcchp1C2djg5AxcOM8BdbaMj0xg7UHrZsDafhI,8013
 newsworthycharts/datawrapper.py,sha256=RRkAVTpfP4updKxUIBaSmKuBi2RUVPaBRF8HDQhlGGA,11250
 newsworthycharts/map.py,sha256=c409jEO4L8Yr780sJRC0RchR44roAlgOUDAkuk1SfRg,6057
@@ -16,7 +16,7 @@ newsworthycharts/custom/climate_cars.py,sha256=WyNLgjgRCv_zRrzVuk_BmcigFSzpzudjE
 newsworthycharts/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 newsworthycharts/lib/color_fn.py,sha256=ck65GzTS9oPgHUCnKDwX2z8zlnNyzncw97MmhXwi8z8,1282
 newsworthycharts/lib/colors.py,sha256=U04TDkvoMQkcldRFXfnwyLOTwq1SWW2se-Ad-DNcw9o,485
-newsworthycharts/lib/datalist.py,sha256=rO7kA0-waYOLMj4AqcksjPWBmn3q1cPyV0xxEUqkukA,6206
+newsworthycharts/lib/datalist.py,sha256=ZyGHnHbn57npITgr2HmtfrS-Xv8U1muCMDOcOwQnlM8,6345
 newsworthycharts/lib/formatter.py,sha256=GNH43hE0bC17OgiV8LYH3YUrEhm7OJh9XzfSV4HVtHo,4838
 newsworthycharts/lib/geography.py,sha256=K0_teFmuPJwXX7Py-amJB_1YY5_gL2kBYhz1LrRCyTg,584
 newsworthycharts/lib/locator.py,sha256=rJHdgiA0LASRzdLINhTeU5zOoq1TCMgge4KOBWSagnM,3041
@@ -28,8 +28,8 @@ newsworthycharts/rc/newsworthy,sha256=yOIZvYS6PG1u19VMcdtfj9vbihKQsey5IprwqK59Kg
 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.65.4.dist-info/LICENSE.txt,sha256=Sq6kGICrehbhC_FolNdXf0djKjTpv3YqjFCIYsxdQN4,1069
-newsworthycharts-1.65.4.dist-info/METADATA,sha256=TfdPxTNdtGU_Rfck2IIlVGWC1CeL7v7ynEZdYkSuvGY,30905
-newsworthycharts-1.65.4.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
-newsworthycharts-1.65.4.dist-info/top_level.txt,sha256=dn_kzIj8UgUCMsh1PHdVEQJHVGSsN7Z8YJF-8xXa8n0,17
-newsworthycharts-1.65.4.dist-info/RECORD,,
+newsworthycharts-1.67.0.dist-info/LICENSE.txt,sha256=Sq6kGICrehbhC_FolNdXf0djKjTpv3YqjFCIYsxdQN4,1069
+newsworthycharts-1.67.0.dist-info/METADATA,sha256=drsWEzZU63lSRO05hy_VT2Mt13UGWyZNW8j_hwI43pc,31262
+newsworthycharts-1.67.0.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
+newsworthycharts-1.67.0.dist-info/top_level.txt,sha256=dn_kzIj8UgUCMsh1PHdVEQJHVGSsN7Z8YJF-8xXa8n0,17
+newsworthycharts-1.67.0.dist-info/RECORD,,