Как изменить цвет двух точек класса на графике 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.