Connect points based on same value in matplotlib

Question

Given this example:

df = pd.DataFrame({'a': ['foo', 'foo', 'bar', 'baz', 'baz', 'baz', 'foobar'], 'x': [3, 4, 1, 5, 5.5, 5.2, 8]})

How can I draw a line plot connecting the data points that have the same 'a', e.g. connect points 3 and 4 because 'foo' is the same, whereas the last data point, 8 is plotted as just a dot, by itself.

Nk03 · Accepted Answer · 2021-06-22 10:08:53Z

1

One way:

df.reset_index().pivot_table(index=  'index', columns= 'a', values = 'x').plot(marker = '*')

Alternative via seaborn:

import seaborn as sns
sns.lineplot(x = 'index', y = 'x' , hue = 'a', data = df.reset_index(), marker = '*')

OUTPUT:

answered Jun 22, 2021 at 10:08

Nk03

15k2 gold badges11 silver badges24 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

user270199 Over a year ago

Thanks. How can I use just one color for the line?

user270199 Over a year ago

One way to do it is to get the unique values of the 'a' column and create a palette where you're assigning the chosen color to each value, then pass it to the palette argument of sns.lineplot().

Collectives™ on Stack Overflow

Connect points based on same value in matplotlib

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related