newsworthycharts 1.70.1__py3-none-any.whl → 1.71.0__py3-none-any.whl

newsworthycharts/__init__.py

@@ -1,10 +1,11 @@
-__version__ = "1.70.1"
+__version__ = "1.71.0"
 
 from .chart import Chart
 from .choroplethmap import ChoroplethMap
 from .bubblemap import BubbleMap
 from .serialchart import SerialChart
 from .seasonalchart import SeasonalChart
+from .rankchart import BumpChart
 from .categoricalchart import CategoricalChart, CategoricalChartWithReference, ProgressChart
 from .scatterplot import ScatterPlot
 from .datawrapper import DatawrapperChart
@@ -26,6 +27,7 @@ CHART_ENGINES = {
     "SeasonalChart": SeasonalChart,
     "SerialChart": SerialChart,
     "StripeChart": StripeChart,
+    "BumpChart": BumpChart,
 
     # custom
     "ClimateCarsYearlyEmissionsTo2030": ClimateCarsYearlyEmissionsTo2030,

newsworthycharts/chart.py

@@ -66,6 +66,7 @@ class Chart(object):
         self.decimals = None
         self.yline = None
         self.type = None
+        self.revert_value_axis = False
         # number of decimals to show in annotations, value ticks, etc
         # None means automatically chose the best number
         self.force_decimals = False
@@ -572,7 +573,7 @@ class Chart(object):
             elif self.units == "percent":
                 # Cut y axis at 100 percent if not value is above
                 value_lim = self.ax.get_ylim() if value_axis == "_y" else self.ax.get_xlim()
-                if self.data.max_val < 1 and value_lim[1] > 1:
+                if self.data.max_val <= 1 and value_lim[1] > 1:
                     value_lim = self.ax.get_ylim() if value_axis == "_y" else self.ax.get_xlim()
                     if value_axis == "_y":
                         self.ax.set_ylim(value_lim[0], 1)
@@ -669,6 +670,19 @@ class Chart(object):
         # print(sub_canvas_height, self._note_rel_height, self._footer_rel_height)
         self._fig.subplots_adjust(bottom=sub_canvas_height)
 
+        if self.revert_value_axis:
+            if hasattr(self, "orientation") and self.orientation == "horizontal":
+                self.ax.invert_xaxis()
+                value_axis = self.ax.get_xaxis()
+                value_ticks = self.ax.get_xticks()
+            else:
+                self.ax.invert_yaxis()
+                value_axis = self.ax.get_yaxis()
+                value_ticks = self.ax.get_yticks()
+            value_ticks = [x for x in value_ticks if x <= self.data.max_val]
+            value_axis.set_ticks(value_ticks)
+            # self.ax.set_ylim(self.data.max_val, self.ymin)
+
     @classmethod
     def init_from(cls, args: dict, storage=LocalStorage(),
                   style: str="newsworthy", language: str='en-GB'):

newsworthycharts/rankchart.py

@@ -0,0 +1,35 @@
+"""
+A chart showing ranking over time (like ”most popular baby names”)
+"""
+from .serialchart import SerialChart
+
+
+class BumpChart(SerialChart):
+    """Plot a rank chart
+
+        Data should be a list of iterables of (rank, date string) tuples, eg:
+    `[ [("2010-01-01", 2), ("2011-01-01", 3)] ]`, combined with a list of
+    labels in the same order
+    """
+
+    def __init__(self, *args, **kwargs):
+        super(BumpChart, self).__init__(*args, **kwargs)
+        self.line_width = 0.5
+        self.label_placement = 'line'
+        self.type = "line"
+        self.decimals = 0
+        self.revert_value_axis = True
+        self.ymin = 1
+        self.allow_broken_y_axis = False
+        self.grid = False
+        self.line_marker = "o-"
+
+    def _get_line_colors(self, i, *args):
+        if not self.data:
+            # Don't waste time
+            return None
+        if self.highlight and self.highlight in self.labels and i == self.labels.index(self.highlight):
+            return self._nwc_style["strong_color"]
+        elif self.colors and i < len(self.colors):
+            return self.colors[i]
+        return self._nwc_style["neutral_color"]

newsworthycharts/serialchart.py

@@ -34,6 +34,10 @@ class SerialChart(Chart):
         # Set with of lines explicitly (otherwise determined by style file)
         self.line_width = None
 
+        self.grid = True
+        self.point_marker = "."
+        self.line_marker = "-"
+
         self.max_ticks = 10
 
         # Manually set tick locations and labels? Provide a list of tuples:
@@ -235,7 +239,10 @@ class SerialChart(Chart):
                 else:
                     lw = self.line_width
 
-                if self.colors is not None:
+                if hasattr(self, "_get_line_colors"):
+                    # Hook for sub classes
+                    color = self._get_line_colors(i)
+                elif self.colors is not None:
                     if self.colors == "qualitative_colors":
                         color = self._nwc_style["qualitative_colors"][i]
                     else:
@@ -245,18 +252,26 @@ class SerialChart(Chart):
                 else:
                     color = self._nwc_style["neutral_color"]
 
-                line, = self.ax.plot(dates, values,
-                                     color=color,
-                                     zorder=zo,
-                                     lw=lw)
+                line, = self.ax.plot(
+                    dates,
+                    values,
+                    self.line_marker,
+                    markersize=4,
+                    color=color,
+                    zorder=zo,
+                    lw=lw,
+                )
                 # Add single, orphaned data points as markers
                 # None, 1, None, 1, 1, 1 =>  . ---
                 num_values = len(values)
                 if num_values == 1:
-                    self.ax.plot(dates[0], values[0],
-                                 c=color,
-                                 marker='.',
-                                 zorder=12)
+                    self.ax.plot(
+                        dates[0],
+                        values[0],
+                        c=color,
+                        marker=self.point_marker,
+                        zorder=12,
+                    )
                 elif num_values > 1:
                     for j, v in enumerate(values):
                         def nullish(val):
@@ -270,10 +285,12 @@ class SerialChart(Chart):
                             elif nullish(values[j - 1]) and nullish(values[j + 1]):
                                 plot_me = True
                         if plot_me:
-                            self.ax.plot(dates[j], v,
-                                         c=color,
-                                         marker='.',
-                                         zorder=12)
+                            self.ax.plot(
+                                dates[j], v,
+                                c=color,
+                                marker=self.point_marker,
+                                zorder=12
+                            )
 
                 if len(self.labels) > i and any([x[1] for x in serie]):
                     line.set_label(self.labels[i])
@@ -559,7 +576,7 @@ class SerialChart(Chart):
                          ymax=ymax + padding_top)
 
         self.ax.yaxis.set_major_formatter(y_formatter)
-        self.ax.yaxis.grid(True)
+        self.ax.yaxis.grid(self.grid)
 
         if ymin > self.baseline and self.allow_broken_y_axis:
             self._mark_broken_axis()

{newsworthycharts-1.70.1.dist-info → newsworthycharts-1.71.0.dist-info}/METADATA

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: newsworthycharts
-Version: 1.70.1
+Version: 1.71.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.70.1.tar.gz
+Download-URL: https://github.com/jplusplus/newsworthycharts/archive/1.71.0.tar.gz
 Author: Jens Finnäs and Leo Wallentin, J++ Stockholm
 Author-email: stockholm@jplusplus.org
 License: MIT
@@ -174,6 +174,12 @@ These settings are available for all chart types:
 
 _ Inherits from SerialChart _
 
+**BumpChart**
+
+_ Inherits from SerialChart _
+
+- highlight: The label value to of a line to highlight
+
 **BubbleMap**
 
 - bubble_size = 1  # The size of the bubbles
@@ -261,6 +267,14 @@ Roadmap
 Changelog
 ---------
 
+- 1.71.0
+
+  - Add `BumpChart`
+
+- 1.70.2
+
+  - Fix edge case bug in y axis cutoff logic
+
 - 1.70.1
 
   - Make sure y axis is cut at 100 % unless we are really, really certain it should be higher

{newsworthycharts-1.70.1.dist-info → newsworthycharts-1.71.0.dist-info}/RECORD

@@ -1,14 +1,15 @@
-newsworthycharts/__init__.py,sha256=CItMiP444DSZFQjCuqz-O39_DrOEA4dtkGMFOMjmons,1160
+newsworthycharts/__init__.py,sha256=mNGMLj9gC8MHheLQftfW3gxIMbVmzuTo9CATDA6O_pw,1221
 newsworthycharts/bubblemap.py,sha256=nkocWmpiFgfjEuJGAsthjY5X7Q56jXWsZHUGXw4PwgE,2587
 newsworthycharts/categoricalchart.py,sha256=YL3pzq8a7U5F0MpeRIAEwCAHw1BRDyKjoaAhf-UyRDo,18179
-newsworthycharts/chart.py,sha256=iMQVbop5DNEAxlG7PlPoUBzZS0j1j19Y4fJFxNgxTrQ,35054
+newsworthycharts/chart.py,sha256=WGrbn4UndhhKq-fbeHMn9vTaEAW3bSHVXrV-8aRqC_8,35692
 newsworthycharts/choroplethmap.py,sha256=Si-01213rWqDKINkhmKV6x5iSMumQveJfrlCnqYcIJM,8173
 newsworthycharts/datawrapper.py,sha256=RRkAVTpfP4updKxUIBaSmKuBi2RUVPaBRF8HDQhlGGA,11250
 newsworthycharts/map.py,sha256=c409jEO4L8Yr780sJRC0RchR44roAlgOUDAkuk1SfRg,6057
 newsworthycharts/rangeplot.py,sha256=NE1W9TnmlpK6T3RvBJOU3nd73EXqkj17OY9i5zlw_cQ,8366
+newsworthycharts/rankchart.py,sha256=G80MO0oF57dgb1Xe3fAQ2hGxU6VhXp_VLev8suhCQjE,1154
 newsworthycharts/scatterplot.py,sha256=l2w2JkA_4WzxQG9XG1Pn62IlJOdCMm1VemMiZL0rEfU,4959
 newsworthycharts/seasonalchart.py,sha256=rr55yqJUkaYDR9Ik98jes6574oY1U8t8LwoLE3gClW4,1967
-newsworthycharts/serialchart.py,sha256=_TBZiOyU-TtnnkJGbP4yZcTlypdIpbxHK4AjE4Su3k8,27787
+newsworthycharts/serialchart.py,sha256=o8a1X_g1lBReOTEKlFM_KiEMXHZ6ImBWqlBpDaDCMAE,28220
 newsworthycharts/storage.py,sha256=myERhlpvXyExXxUByBq9eW1bWkCyfH9SwTZbsWSyy3Q,4301
 newsworthycharts/stripechart.py,sha256=9B6PX2MyLuKNQ8W0OGdKbP0-U32kju0K_NHHwwz_J68,1547
 newsworthycharts/custom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -28,8 +29,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.70.1.dist-info/LICENSE.txt,sha256=Sq6kGICrehbhC_FolNdXf0djKjTpv3YqjFCIYsxdQN4,1069
-newsworthycharts-1.70.1.dist-info/METADATA,sha256=bSFlrdFHV5xU68LB81ZEMUV5ypwD8NpfqMV4yNN--EU,32033
-newsworthycharts-1.70.1.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
-newsworthycharts-1.70.1.dist-info/top_level.txt,sha256=dn_kzIj8UgUCMsh1PHdVEQJHVGSsN7Z8YJF-8xXa8n0,17
-newsworthycharts-1.70.1.dist-info/RECORD,,
+newsworthycharts-1.71.0.dist-info/LICENSE.txt,sha256=Sq6kGICrehbhC_FolNdXf0djKjTpv3YqjFCIYsxdQN4,1069
+newsworthycharts-1.71.0.dist-info/METADATA,sha256=Wcl3NZM1kJhK7HiHsongNyXTmVeOj0iDzMIBThn2U1k,32222
+newsworthycharts-1.71.0.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
+newsworthycharts-1.71.0.dist-info/top_level.txt,sha256=dn_kzIj8UgUCMsh1PHdVEQJHVGSsN7Z8YJF-8xXa8n0,17
+newsworthycharts-1.71.0.dist-info/RECORD,,