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

Question

我必须发出许多绘图命令来创建图片。如下例所示：

.meta

为简化代码，我更喜欢以下内容：

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

这可能吗？

Answer 1

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

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

要获得当前颜色周期的第一种颜色，您可以使用

{{1}}