Add _skipnan to plotlines to allow joining two points with a line

backtrader · backtrader · commit 0b8d6aea7ba0 · 2018-01-28T14:57:54.000+01:00
diff --git a/backtrader/plot/plot.py b/backtrader/plot/plot.py
@@ -438,7 +438,18 @@ def plotind(self, iref, ind,
                 plotkwargs['zorder'] = self.pinf.zordernext(ax)
 
             pltmethod = getattr(ax, lineplotinfo._get('_method', 'plot'))
-            plottedline = pltmethod(self.pinf.xdata, lplot, **plotkwargs)
+
+            xdata, lplotarray = self.pinf.xdata, lplot
+            if lineplotinfo._get('_skipnan', False):
+                # Get the full array and a mask to skipnan
+                lplotarray = np.array(lplot)
+                lplotmask = np.isfinite(lplotarray)
+
+                # Get both the axis and the data masked
+                lplotarray = lplotarray[lplotmask]
+                xdata = np.array(xdata)[lplotmask]
+
+            plottedline = pltmethod(xdata, lplotarray, **plotkwargs)
             try:
                 plottedline = plottedline[0]
             except:
diff --git a/changelog.txt b/changelog.txt
@@ -1,3 +1,6 @@
+1.9.61.122
+  - Add `_skipnan` to plotlines to allow joining two points with a line
+
 1.9.60.122
   - Remove unused files
   - README update, Docstring corrections, documentation corrections
