CODEWITHSUNDEEP
pythonpandasmatplotlibseaborn
Alexis Eggermont
Alexis Eggermont

Reputation: 8145

Line color depending on variable

I have a pandas dataframe with three columns, Date(timestamp), Color('red' or 'blue') and Value(int).

I am currently getting a line chart from it with the following code:

import matplotlib.pyplot as plt
import pandas as pd
Dates=['01/01/2014','02/01/2014','03/01/2014','04/01/2014','05/01/2014','06/01/2014','07/01/2014']
Values=[3,4,6,5,4,5,4]
Colors=['red','red','blue','blue','blue','red','red']
df=pd.DataFrame({'Dates':Dates,'Values':Values,'Colors':Colors})
df['Dates']=pd.to_datetime(df['Dates'],dayfirst=True)

grouped = df.groupby('Colors')
fig, ax = plt.subplots()

for key, group in grouped:
   group.plot(ax=ax, x="Dates", y="Values", label=key, color=key)

plt.show()

I'd like the line color to depend on the 'color' columns. How can I achieve that?

I have seen here a similar question for scatterplots, but it doesn't seem I can apply the same solution to a time series line chart.

My output is currently this:

enter image description here

I am trying to achieve something like this (one line only, but several colors)

enter image description here

Upvotes: 3

Views: 13160

Answers (1)

Anton Protopopov
Anton Protopopov

Reputation: 31662

As I said you could find the answer from the link I attached in the comment:

Dates = ['01/01/2014', '02/01/2014', '03/01/2014', '03/01/2014', '04/01/2014', '05/01/2014']
Values = [3, 4, 6, 6, 5, 4]
Colors = ['red', 'red', 'red', 'blue', 'blue', 'blue']
df = pd.DataFrame({'Dates': Dates, 'Values': Values, 'Colors': Colors})
df['Dates'] = pd.to_datetime(df['Dates'], dayfirst=True)

grouped = df.groupby('Colors')
fig, ax = plt.subplots(1)

for key, group in grouped:
    group.plot(ax=ax, x="Dates", y="Values", label=key, color=key)

When color changing you need to add extra point to make line continuous

enter image description here

Upvotes: 4

Related Questions