Merge pull request #106 from better/reset-color-cycle

reset color cycle between plots, for multiple models
better · Aug 15, 2019 · d6187b6 · d6187b6
2 parents 64c539d + ce58fbc
commit d6187b6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/convoys/plotting.py b/convoys/plotting.py
@@ -69,6 +69,7 @@ def plot_cohorts(G, B, T, t_max=None, model='kaplan-meier',
     # Plot
     t = numpy.linspace(0, t_max, 1000)
     _, y_max = ax.get_ylim()
+    ax.set_prop_cycle(None)  # Reset to first color
     for i, group in enumerate(specific_groups):
         j = groups.index(group)  # matching index of group