Update lib/matplotlib/colorbar.py

oscargus · timhoffm · web-flow · commit c6729ba3c033 · 2022-06-14T10:06:09.000+02:00
Co-authored-by: Tim Hoffmann &lt;2836374+timhoffm@users.noreply.github.com&gt;
diff --git a/lib/matplotlib/colorbar.py b/lib/matplotlib/colorbar.py
@@ -653,10 +653,8 @@ def _add_solids(self, X, Y, C):
                     self.solids.set_rasterized(True)
         if self.drawedges:
             start_idx = 0 if self._extend_lower() else 1
-            if self._extend_upper():
-                self.dividers.set_segments(np.dstack([X, Y])[start_idx:])
-            else:
-                self.dividers.set_segments(np.dstack([X, Y])[start_idx:-1])
+            end_idx = len(X) if self._extend_upper() else -1
+            self.dividers.set_segments(np.dstack([X, Y])[start_idx:end_idx])
         else:
             self.dividers.set_segments([])
 
