web-dev-qa-db-fra.com

Puis-je tracer une ligne de régression et afficher les paramètres à l'aide du nuage de points avec une trame de données pandas?

Je voudrais produire un diagramme de dispersion à partir d'une trame de données Pandas en utilisant le code suivant:

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

Y a-t-il un paramètre que je peux envoyer avec la déclaration, donc il trace une ligne de régression et montre les paramètres de l'ajustement?

quelque chose comme:

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

Je ne pense pas qu'il existe un tel paramètre pour DataFrame.plot (). Cependant, vous pouvez facilement y parvenir en utilisant Seaborn . Passez simplement le pandas dataframe à lmplot (en supposant que vous avez installé seaborn):

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