Pearson correlation

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

calc314
4191 ●22 ●48 ●113

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

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 ( 2012-10-13 18:19:27 +0100 )edit

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

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

I thought R came with the standard Sage installation.

calc314 ( 2012-10-13 23:06:22 +0100 )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 ( 2012-10-14 00:10:49 +0100 )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 ( 2012-10-14 00:36:42 +0100 )edit

see more comments

1 Answer

Comments

Your Answer

Question Tools

Stats

Related questions

Pearson correlation edit

1 Answer

Comments