2016-04-05 21 views
5

Vorrei produrre un grafico a dispersione da un dataframe Panda utilizzando il seguente codice:Posso disegnare una linea di regressione e mostrare i parametri utilizzando il grafico a dispersione con un dataframe panda?

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

C'è un parametro posso inviare con la dichiarazione, in modo che traccia una linea di regressione e mostra i parametri della forma?

qualcosa di simile:

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

risposta

12

Non credo che ci sia una tale paramter per DataFrame.plot(). Tuttavia, puoi facilmente ottenerlo utilizzando Seaborn. Basta passare il dataframe panda a lmplot (ammesso che abbiate installato Seaborn):

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

grande! per me funziona. Sai come posso tracciare i parametri di Regressione sulla Mappa? –

+0

Sfortunatamente, questo sembra non essere possibile usando lmplot come postato in questa [domanda] (http://stackoverflow.com/questions/22852244/how-to-get-the-numerical-fitting-results-when-plotting-a -regression-in-Seaborn). Tuttavia, si può dare un'occhiata a questo problema su [github] (https://github.com/mwaskom/seaborn/issues/207). –

+0

Grazie mille per il vostro aiuto. –

Problemi correlati