Как изменить цвет линии после вызова графика matplotlib?

Question

Следующий код, взятый из учебника thetechrepo, отобразит некоторые данные с разными цветами.

import matplotlib.pyplot as plt 
plt.figure() 

#create data 
x_series = [0,1,2,3,4,5] 
y_series_1 = [x**2 for x in x_series] 
y_series_2 = [x**3 for x in x_series] 

plt.plot(x_series, y_series_1, 'r-') 
plt.plot(x_series, y_series_2, 'c--') 
plt.show() 

Однако то, что если бы я хотел изменить цвета после я назвал сюжет? Например, как изменить цвет series_1 на зеленый после того, как я позвонил plt.plot(x_series, y_series_1, 'r-')?

Answer 1

Вы можете использовать set_color на Line2D объект, созданный plt.plot. Например:

l1, = plt.plot(x_series, y_series_1, 'r-') 
l2, = plt.plot(x_series, y_series_2, 'c--') 

# Some time later... 
l1.set_color('b') 
l2.set_color('g')