FIX Explicitly specify the plot colors for each curve

Author: raghavrv

examples/linear_model/plot_lasso_coordinate_descent_path.py

@@ -13,6 +13,7 @@
 # Author: Alexandre Gramfort <[email protected]>
 # License: BSD 3 clause
 
+from itertools import cycle
 import numpy as np
 import matplotlib.pyplot as plt
 
@@ -47,9 +48,13 @@
 
 plt.figure(1)
 ax = plt.gca()
-ax.set_color_cycle(2 * ['b', 'r', 'g', 'c', 'k'])
-l1 = plt.plot(-np.log10(alphas_lasso), coefs_lasso.T)
-l2 = plt.plot(-np.log10(alphas_enet), coefs_enet.T, linestyle='--')
+
+colors = cycle(['b', 'r', 'g', 'c', 'k'])
+neg_log_alphas_lasso = -np.log10(alphas_lasso)
+neg_log_alphas_enet = -np.log10(alphas_enet)
+for coef_l, coef_e, c in zip(coefs_lasso, coefs_enet, colors):
+    l1 = plt.plot(neg_log_alphas_lasso, coef_l, c=c)
+    l2 = plt.plot(neg_log_alphas_enet, coef_e, linestyle='--', c=c)
 
 plt.xlabel('-Log(alpha)')
 plt.ylabel('coefficients')
@@ -60,10 +65,10 @@
 
 plt.figure(2)
 ax = plt.gca()
-ax.set_color_cycle(2 * ['b', 'r', 'g', 'c', 'k'])
-l1 = plt.plot(-np.log10(alphas_lasso), coefs_lasso.T)
-l2 = plt.plot(-np.log10(alphas_positive_lasso), coefs_positive_lasso.T,
-              linestyle='--')
+neg_log_alphas_positive_lasso = -np.log10(alphas_positive_lasso)
+for coef_l, coef_pl, c in zip(coefs_lasso, coefs_positive_lasso, colors):
+    l1 = plt.plot(neg_log_alphas_lasso, coef_l, c=c)
+    l2 = plt.plot(neg_log_alphas_positive_lasso, coef_pl, linestyle='--', c=c)
 
 plt.xlabel('-Log(alpha)')
 plt.ylabel('coefficients')
@@ -74,10 +79,10 @@
 
 plt.figure(3)
 ax = plt.gca()
-ax.set_color_cycle(2 * ['b', 'r', 'g', 'c', 'k'])
-l1 = plt.plot(-np.log10(alphas_enet), coefs_enet.T)
-l2 = plt.plot(-np.log10(alphas_positive_enet), coefs_positive_enet.T,
-              linestyle='--')
+neg_log_alphas_positive_enet = -np.log10(alphas_positive_enet)
+for (coef_e, coef_pe, c) in zip(coefs_enet, coefs_positive_enet, colors):
+    l1 = plt.plot(neg_log_alphas_enet, coef_e, c=c)
+    l2 = plt.plot(neg_log_alphas_positive_enet, coef_pe, linestyle='--', c=c)
 
 plt.xlabel('-Log(alpha)')
 plt.ylabel('coefficients')

examples/linear_model/plot_ridge_path.py

@@ -47,7 +47,6 @@
 # Display results
 
 ax = plt.gca()
-ax.set_color_cycle(['b', 'r', 'g', 'c', 'k', 'y', 'm'])
 
 ax.plot(alphas, coefs)
 ax.set_xscale('log')