Tengo un df que se ve a continuación:

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

Quiero tener un diagrama donde el eje Y es el eje A y X la fecha, y el problema es con el color. Quiero todas las ocurrencias de 76 en rojo, 32 en azul y todos los demás valores de A en color verde. es posible?

1
hakuna_code 2 oct. 2019 a las 14:25

1 respuesta

La mejor respuesta

Sí, puedes hacerlo:

# 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)

Salida:

enter image description here

1
Quang Hoang 2 oct. 2019 a las 13:24