如何在python绘图中更改两个类点的颜色？

Question

当我使用ax.plot函数时，我想改变点的颜色

groups = error_df.groupby('true_class')
fig, ax = plt.subplots()
for name, group in groups:
    ax.plot(group.index, group.reconstruction_error, marker='o', ms=4, linestyle='','g','k',
            label= "fraud" if name == 1 else "normal")

根据代码，该图有两个类点“欺诈”和“正常”。

“欺诈”是橙色的，“正常”是蓝色的。

我想把“欺诈”的颜色改成绿色，把“正常”改成红色。

我该怎么做呢？谢谢。

Answer 1

看起来您应该能够像使用label一样执行此操作，即通过

color="green" if name == 1 else "red"

作为plot方法的参数。