Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

How do I create a multiline plot using seaborn?

I am trying out Seaborn to make my plot visually better than matplotlib. I have a dataset which has a column 'Year' which I want to plot on the X-axis and 4 Columns say A,B,C,D on the Y-axis using different coloured lines. I was trying to do this using the sns.lineplot method but it allows for only one variable on the X-axis and one on the Y-axis. I tried doing this

sns.lineplot(data_preproc['Year'],data_preproc['A'], err_style=None) sns.lineplot(data_preproc['Year'],data_preproc['B'], err_style=None) sns.lineplot(data_preproc['Year'],data_preproc['C'], err_style=None) sns.lineplot(data_preproc['Year'],data_preproc['D'], err_style=None) 

But this way I don't get a legend in the plot to show which coloured line corresponds to what. I tried checking the documentation but couldn't find a proper way to do this.

like image 885
SSPdude Avatar asked Sep 13 '18 07:09

SSPdude


People also ask

How do you make a multiple line plot in Seaborn?

You probably need to re-organize your dataframe in a suitable way so that there is one column for the x data, one for the y data, and one which holds the label for the data point. You can also just use matplotlib. pyplot . If you import seaborn , much of the improved design is also used for "regular" matplotlib plots.

How do you plot multiple columns in Seaborn?

We can make multiple columns of the barplot by using the seaborn function group bar. The groupby() method in Pandas is used to divide data into groups depending on specified criteria. In the following example script, we have included the matplotlib library and seaborn module for plotting multiple columns using barplot.


2 Answers

Seaborn favors the "long format" as input. The key ingredient to convert your DataFrame from its "wide format" (one column per measurement type) into long format (one column for all measurement values, one column to indicate the type) is pandas.melt. Given a data_preproc structured like yours, filled with random values:

num_rows = 20 years = list(range(1990, 1990 + num_rows)) data_preproc = pd.DataFrame({     'Year': years,      'A': np.random.randn(num_rows).cumsum(),     'B': np.random.randn(num_rows).cumsum(),     'C': np.random.randn(num_rows).cumsum(),     'D': np.random.randn(num_rows).cumsum()}) 

A single plot with four lines, one per measurement type, is obtained with

sns.lineplot(x='Year', y='value', hue='variable',               data=pd.melt(data_preproc, ['Year'])) 

(Note that 'value' and 'variable' are the default column names returned by melt, and can be adapted to your liking.)

like image 192
dnswlt Avatar answered Oct 20 '22 22:10

dnswlt


See the documentation:

sns.lineplot(x="Year", y="signal", hue="label", data=data_preproc) 

You probably need to re-organize your dataframe in a suitable way so that there is one column for the x data, one for the y data, and one which holds the label for the data point.

You can also just use matplotlib.pyplot. If you import seaborn, much of the improved design is also used for "regular" matplotlib plots. Seaborn is really "just" a collection of methods which conveniently feed data and plot parameters to matplotlib.

like image 22
IonicSolutions Avatar answered Oct 20 '22 23:10

IonicSolutions