如何设置后续绘图命令的颜色？

Question

我必须发出很多绘图命令才能创建图片。如下面的例子：

color="C0-" 
plt.plot([1,5,4], color) 
plt.plot([3,7,8], color) 

为了简化代码，我宁愿是这样的：

plt.set_color("C0-") # ERROR 
plt.plot([1,5,4]) 
plt.plot([3,7,8]) 

这可能吗？

Answer 1

如果未给出颜色参数，则图的颜色由正在使用的颜色循环决定。您可以在运行时更改颜色周期以仅使用一种颜色，这有效地使所有后续图表具有相同的颜色。

plt.gca().set_prop_cycle('color', ["blue"]) 

为了得到当前颜色周期的第一个颜色，你可以使用

c0 = plt.rcParams["axes.prop_cycle"].by_key()["color"][0] 
plt.gca().set_prop_cycle('color', [c0])