CODEWITHSUNDEEP
pythonmatplotlibseaborn
user2489252
user2489252

Reputation:

Scatterplot without linear fit in seaborn

I am wondering if there is a way to turn off the linear fit in seaborn's lmplot or if there is an equivalent function that just produces the scatterplot. Sure, I could also use matplotlib, however, I find the syntax and aesthetics in seaborn quite appealing. E.g,. I want to plot the following plot

import seaborn as sns
sns.set(style="ticks")

df = sns.load_dataset("anscombe")
sns.lmplot("x", "y", data=df, hue='dataset')

enter image description here

Without the linear fit like so:

from itertools import cycle
import numpy as np

import matplotlib.pyplot as plt

color_gen = cycle(('blue', 'lightgreen', 'red', 'purple', 'gray', 'cyan'))

for lab in np.unique(df['dataset']):
    plt.scatter(df.loc[df['dataset'] == lab, 'x'], 
                df.loc[df['dataset'] == lab, 'y'], 
                c=next(color_gen),
                label=lab)

plt.legend(loc='best')

enter image description here

Upvotes: 24

Views: 31202

Answers (3)

Rudra Mohan
Rudra Mohan

Reputation: 790

I recommend instead of sns.lmplot() to use sns.scatterplot()

# import libaries
import seaborn as sns

# load tips dataset from GitHub seaborn repository
tips_df = sns.load_dataset("tips")

#create scatter plot
sns.scatterplot(x = "tip", y = "total_bill", data = tips_df, hue ="sex")

To learn more in detail follow seaborn scatter plot using sns.scatterplot() tutorial

Upvotes: -2

Michael Hall
Michael Hall

Reputation: 3330

This doesn't directly answer the question, but may help others who find there way here who just want to do a plain old scatter plot.
As of version 0.9.0 seaborn now has a scatterplot method.

import seaborn as sns
sns.set(style="ticks")

df = sns.load_dataset("anscombe")
sns.scatterplot("x", "y", data=df, hue='dataset')

enter image description here

Upvotes: 1

HYRY
HYRY

Reputation: 97261

set fit_reg argument to False:

sns.lmplot("x", "y", data=df, hue='dataset', fit_reg=False)

Upvotes: 48

Related Questions