Re: [Rd] Suggestion to extend aggregate() to return multiple and/or named values

From: Mike Lawrence <Mike.Lawrence_at_DAL.CA>
Date: Fri, 13 Jul 2007 14:46:01 -0300

bugfix already :P prior version fails when there is only one factor in Ind. This version also might be faster as I avoid using aggregate to create the dummy frame.

agg=function(z,Ind,FUN,...){

	FUN.out=by(z,Ind,FUN,...)
	num.cells=length(FUN.out)
	num.values=length(FUN.out[[1]])
	
	for(i in 1:length(Ind)){
		Ind[[i]]=unique(Ind[[i]])
	}
	temp=expand.grid(Ind)

	for(i in 1:num.values){
		temp$new=NA
		n=names(FUN.out[[1]])[i]
		names(temp)[length(temp)]=ifelse(!is.null(n),n,ifelse(i==1,'x',paste 
('x',i,sep='')))
		for(j in 1:num.cells){
			temp[j,length(temp)]=FUN.out[[j]][i]
		}
	}
	return(temp)

}

On 13-Jul-07, at 1:29 PM, Mike Lawrence wrote:

> Hi all,
>
> This is my first post to the developers list. As I understand it,
> aggregate() currently repeats a function across cells in a
> dataframe but is only able to handle functions with single value
> returns. Aggregate() also lacks the ability to retain the names
> given to the returned value. I've created an agg() function (pasted
> below) that is apparently backwards compatible (i.e. returns
> identical results as aggregate() if the function returns a single
> unnamed value), but is able to handle named and/or multiple return
> values. The code may be a little inefficient (there must be an
> easier way to set up the 'temp' data frame than to call aggregate
> and remove the final column), but I'm suggesting that something
> similar to this may be profitably used to replace aggregate entirely.
>
> #modified aggregate command, allowing for multiple/named output values
> agg=function(z,Ind,FUN,...){
> FUN.out=by(z,Ind,FUN,...)
> num.cells=length(FUN.out)
> num.dv=length(FUN.out[[1]])
>
> temp=aggregate(z,Ind,length) #dummy data frame
> temp=temp[,c(1:(length(temp)-1))] #remove last column from dummy
> frame
>
> for(i in 1:num.dv){
> temp=cbind(temp,NA)
> n=names(FUN.out[[1]])[i]
> names(temp)[length(temp)]=ifelse(!is.null(n),n,ifelse
> (i==1,'x',paste('x',i,sep='')))
> for(j in 1:num.cells){
> temp[j,length(temp)]=FUN.out[[j]][i]
> }
> }
> return(temp)
> }
>
> #create some factored data
> z=rnorm(100) # the DV
> A=rep(1:2,each=25,2) #one factor
> B=rep(1:2,each=50) #another factor
> Ind=list(A=A,B=B) #the factor list
>
> aggregate(z,Ind,mean) #show the means of each cell
> agg(z,Ind,mean) #should be identical to aggregate
>
> aggregate(z,Ind,summary) #returns an error
> agg(z,Ind,summary) #returns named columns
>
> #Make a function that returns multiple unnamed values
> summary2=function(x){
> s=summary(x)
> names(s)=NULL
> return(s)
> }
> agg(z,Ind,summary2) #returns multiple columns, default names
>
>
> --
> Mike Lawrence
> Graduate Student, Department of Psychology, Dalhousie University
>
> Website: http://memetic.ca
>
> Public calendar: http://icalx.com/public/informavore/Public
>
> "The road to wisdom? Well, it's plain and simple to express:
> Err and err and err again, but less and less and less."
> - Piet Hein
>
>

--
Mike Lawrence
Graduate Student, Department of Psychology, Dalhousie University

Website: http://memetic.ca

Public calendar: http://icalx.com/public/informavore/Public

"The road to wisdom? Well, it's plain and simple to express:
Err and err and err again, but less and less and less."
	- Piet Hein

______________________________________________
R-devel_at_r-project.org mailing list
https://stat.ethz.ch/mailman/listinfo/r-devel
Received on Fri 13 Jul 2007 - 17:51:30 GMT

Archive maintained by Robert King, hosted by the discipline of statistics at the University of Newcastle, Australia.
Archive generated by hypermail 2.2.0, at Fri 13 Jul 2007 - 21:36:40 GMT.

Mailing list information is available at https://stat.ethz.ch/mailman/listinfo/r-devel. Please read the posting guide before posting to the list.