diff --git a/pseudopy/nonnormal.py b/pseudopy/nonnormal.py index 43b8af9..45fb7b7 100644 --- 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 index b79ba19..7a41c72 100644 --- 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