Как изобразить определенные цвета для диапазона значений в кадре данных Python?

Question

У меня есть df, который выглядит следующим образом:

S.No  Date          A   

0     12/07/03     76
1     12/07/13     1
2     12/07/23     32
3     12/08/03     12
4     12/08/04     22 
5     12/08/05     11

Я хочу иметь график, где ось Y - это A, а ось X - дата, и проблема в цвете. Я хочу, чтобы все 76 составляли красным, 32 - синим, а все остальные значения - зеленым. Возможно ли это?

Answer 1

Да, вы можете сделать это:

# define the color according to the values of df['A']
colors = np.select((df['A'].eq(76), df['A'].eq(32)), ('r','b'), 'g')

# pass the color to plt.scatter
plt.scatter(x=df['Date'],y=df['A'], c=colors)

Вывод: