Plotly line plot

Guys i have a problem putting color on my line graph in plotly. I have a data frame and when i choose one columns to use as color i have a rough link of the color values.

my code below:

fig = px.line(n_df, x=n_df.index, y=‘wait’,
color = ‘imp_charge’, hover_data = n_df.iloc[:,:-9])
fig.update_layout(xaxis_title=‘Date’)
fig.show()

I instead wanted something like this: