Ask Your Question
0

Pearson correlation

asked 2012-10-13 13:47:25 +0200

Jaakko Seppälä gravatar image

Is there a method in Sage to compute Pearson correlation of a finite set? I have data points $(x_1,y_1),\ldots,(x_6,y_6)$ and wondering if there is a shorter way to do it than just write one long expression.

edit retag flag offensive close merge delete

1 Answer

Sort by » oldest newest most voted
2

answered 2012-10-13 14:38:56 +0200

calc314 gravatar image

You can call R from within Sage as follows. The command r.cor does require that you give it two lists containing the coordinates from your data set. In the example below, I'll use a list comprehension in Python to do that.

data=[[1,3],[1.5,2],[5,7]]
xdata=[a[0] for a in data]
ydata=[a[1] for a in data]
r.cor(xdata,ydata)

You can also get the regression line easily.

var('a,b')
f(x)=a*x+b
ans=find_fit(data,f)
g(x)=ans[0].rhs()*x+ans[1].rhs()
plot(g(x),(x,0,10))+list_plot(data)
edit flag offensive delete link more

Comments

Instead of doing two list comps, you could use `zip` and unpacking: `x, y = zip(*data)`, for example. (`r.cor(*zip(*data))` would do it in one line, but that looks a little too cute to use.)

DSM gravatar imageDSM ( 2012-10-13 18:19:27 +0200 )edit

Hmm. `Unable to start r`. I downloaded Sage from its WWW-site and installed that.

Jaakko Seppälä gravatar imageJaakko Seppälä ( 2012-10-13 19:09:41 +0200 )edit

I thought R came with the standard Sage installation.

calc314 gravatar imagecalc314 ( 2012-10-13 23:06:22 +0200 )edit

Alternatively, you can use `scipy`'s: `import scipy.stats` and then `scipy.stats.pearsonr(xdata, ydata) ` -- type `help(scipy.stats.pearsonr)` after the import for the details.

DSM gravatar imageDSM ( 2012-10-14 00:10:49 +0200 )edit

I actually like the plots from stats in scipy better than in R for use in Sage. I'm surprised by the `unable to start r` error with r.cor. I've gotten it when using R's graphics in Sage but not with the basic stats commands.

calc314 gravatar imagecalc314 ( 2012-10-14 00:36:42 +0200 )edit

Your Answer

Please start posting anonymously - your entry will be published after you log in or create a new account.

Add Answer

Question Tools

Stats

Asked: 2012-10-13 13:47:25 +0200

Seen: 1,625 times

Last updated: Dec 20 '12