Can I draw a regression line and show parameters using scatterplot with a pandas dataframe?

python pandas regression scatter-plot
Markus W picture Markus W · Apr 5, 2016 · Viewed 23k times · Source

I would like to produce a Scatterplot from a Pandas dataframe using the following code:

df.plot.scatter(x='one', y='two, title='Scatterplot')

Is there a Parameter I can send with the Statement, so it plots a Regression line and shows the Parameters of the fit?

something like:

df.plot.scatter(x='one', y='two', title='Scatterplot', Regression_line)

Answer

Pascal dB picture Pascal dB · Apr 5, 2016

I don't think that there's such a paramter for DataFrame.plot(). However, you can easily achieve this using Seaborn. Just pass the pandas dataframe to lmplot (assuming you have seaborn installed):

import seaborn as sns
sns.lmplot(x='one',y='two',data=df,fit_reg=True)

Related questions

Run an OLS regression with Pandas Data Frame

I have a pandas data frame and I would like to able to predict the values of column A from the values in columns B and C. Here is a toy example: import pandas as pd df = pd.DataFrame({"A": [10,20,30,40,50], "…

Read More
Fixed effect in Pandas or Statsmodels

Is there an existing function to estimate fixed effect (one-way or two-way) from Pandas or Statsmodels. There used to be a function in Statsmodels but it seems discontinued. And in Pandas, there is something called plm, but I can't import …

Read More
ValueError: feature_names mismatch: in xgboost in the predict() function

I have trained an XGBoostRegressor model. When I have to use this trained model for predicting for a new input, the predict() function throws a feature_names mismatch error, although the input feature vector has the same structure as the …

Read More