[R] vectorized nested loop: apply a function that takes two rows

Jose Quesada quesada at gmail.com
Tue Jan 23 21:40:50 CET 2007


Dear R users,

I want to apply a function that takes two vectors as input to all pairs
(combinations (nrow(X), 2))of matrix rows in a matrix.
I know that ideally, one should avoid loops in R, but after reading the docs for
do.call, apply, etc, I still don't know how to write the nested loop in a
vectorized way.

Example data:
x 		= matrix(rnorm(1000), 100, 100)
zeros	= runif(90)
x[]
# this is actually a very large sparse matrix, but it doesn't matter for the
# example
library(Matrix)
x = as(dat,"CsparseMatrix")

# cosine function
cosine = function (x, y){
	if (is.vector(x) && is.vector(y)) {
		return(crossprod(x, y)/sqrt(crossprod(x) * crossprod(y)))
	} else {stop("cosine: argument mismatch. Two vectors needed as input.")}
}

# The loop-based solution I have is:
		if (is(x, "Matrix") ) {
			cos 	= array(NA, c(ncol(x), ncol(x))) # preallocate memory
			for (i in 2:ncol(x)) {
				for (j in 1:(i - 1)) {
					cos[i, j] = cosine(x[, i], x[, j])
				}
			}
		}

This solution seems inneficient. Is there an easy way of achieving this with a
clever do.call + apply combination?
-- 
Thanks in advance,
-Jose

--
Jose Quesada, PhD
Research fellow, Psychology Dept.
Sussex University, Brighton, UK
http://www.andrew.cmu.edu/~jquesada



More information about the R-help mailing list