Merge pull request #7 from andrenarchy/use-prop-cycle

use prop_cycle instead of color_cycle
andrenarchy · Nov 29, 2018 · 61efca9 · 61efca9
2 parents 4c5c529 + ea83c80
commit 61efca9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/pseudopy/nonnormal.py b/pseudopy/nonnormal.py
@@ -119,7 +119,8 @@ def __init__(self, A,
     def plot(self, epsilons, **kwargs):
         contours = pyplot.contour(self.Real, self.Imag, self.Vals,
                                   levels=epsilons,
-                                  colors=pyplot.rcParams['axes.color_cycle'])
+                                  colors=pyplot.rcParams['axes.prop_cycle'].by_key()['color']
+                                  )
         plot_finish(contours, **kwargs)
         return contours
 

diff --git a/pseudopy/normal.py b/pseudopy/normal.py
@@ -23,7 +23,7 @@ def plot(self, epsilons, **kwargs):
                 Y += list(numpy.imag(path.vertices[:-1]))
                 Z += [epsilon] * (len(path.vertices) - 1)
         contours = pyplot.tricontour(X, Y, Z, levels=epsilons,
-                                     colors=pyplot.rcParams['axes.color_cycle']
+                                     colors=pyplot.rcParams['axes.prop_cycle'].by_key()['color']
                                      )
         plot_finish(contours, **kwargs)
         return contours