Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

How can I assign a variable's value to column name in plyr?

Is there any way to assign a variable's value to the resultant column name in plyr? So in this code...

column_name <- 'total'
df <- data.frame(a=c('a','b'), b=c(1,2))
ddply(df, .(a), summarise, column_name=sum(b))

As you know, this spits out a data frame which consists of variables a and column_name. However, what I want to get it is variables a and total, where total is assigned dynamically by a variable's value, since actually, I want to process it inside a loop, that I can't specify it directly inside ddply function.

For you information, this code doesn't work.

ddply(df, .(a), summarise, get(column_name)=sum(b))

So is there any solution to deal with it?

like image 501
Blaszard Avatar asked May 09 '13 05:05

Blaszard


1 Answers

There are several ways:

> column_names <- c('total', 'latot')
> df <- data.frame(a=c('a','b'), b=c(1,2))
> 
> # temporal variable
> for (cn in column_names) {
+   ret <- ddply(df, .(a), summarise, ..x=sum(b))
+   ret <- rename(ret, c(..x = cn))
+   print(ret)
+ }
  a total
1 a     1
2 b     2
  a latot
1 a     1
2 b     2
> 
> # ept solution
> for (cn in column_names) {
+   print(eval(parse(text = paste0("ret <- ddply(df, .(a), summarise,", cn, "=sum(b))"))))
+ }
  a total
1 a     1
2 b     2
  a latot
1 a     1
2 b     2
> 
> # dynamic generation of call
> for (cn in column_names) {
+   args <- alist(df, .(a), summarize, sum(b))
+   names(args) <- c("", "", "", cn)
+   print(do.call("ddply", args))   
+ }
  a total
1 a     1
2 b     2
  a latot
1 a     1
2 b     2
like image 181
kohske Avatar answered Oct 21 '22 06:10

kohske