writing the outcome of a nested loop to a vector object in R

Question

I have the following data read into R as a data frame named "data_old":

   yes year month
1  15 2004     5
2   9 2005     6
3  15 2006     3
4  12 2004     5
5  14 2005     1
6  15 2006     7
.   .  ...     .
.   .  ...     .

I have written a small loop which goes through the data and sums up the yes variable for each month/year combination:

year_f <- c(2004:2006)
month_f <- c(1:12)

for (i in year_f){
    for (j in month_f){
        x <- subset(data_old, month == j & year == i, select="yes")
        if (nrow(x) > 0){
            print(sum(x))
            }
        else{print("Nothing")}
        }
    }

My question is this: I can print the sum for each month/year combination in the terminal, but how do i store it in a vector? (the nested loop is giving me headaches trying to figure this out).

Thomas

score 7 · Accepted Answer · answered Aug 19 '10 at 11:07

7

Another way,

library(plyr)
ddply(data_old,.(year,month),function(x) sum(x[1]))

  year month V1
1 2004     5 27
2 2005     1 14
3 2005     6  9
4 2006     3 15
5 2006     7 15

answered Aug 19 '10 at 11:07

gd047

29,749
18
107
146

3

or `ddply(data_old,.(year,month),summarize, yes = sum(yes))` – JoFrhwld Aug 19 '10 at 13:13

score 6 · Answer 2 · edited May 23 '17 at 12:18

6

Forget the loops, you want to use an aggregation function. There's a recent discussion of them in this SO question.

with(data_old, tapply(yes, list(year, month), sum))

is one of many solutions.

Also, you don't need to use c() when you aren't concatenating anything. Plain 1:12 is fine.

edited May 23 '17 at 12:18

Community

1
1

answered Aug 19 '10 at 10:55

Richie Cotton

118,240
47
247
360

score 3 · Answer 3 · answered Aug 19 '10 at 13:09

3

Just to add a third option:

aggregate(yes ~ year + month, FUN=sum, data=data_old)

answered Aug 19 '10 at 13:09

rcs

67,191
22
172
153

IMO, this is the way it should be done. It's clearer to the average programmer: we are aggregating, not "ddplying". – Vince Aug 19 '10 at 16:28
1

It's a good way to solve this problem, but my brain doesn't fit all of the options for other problems. The nice thing about plyr is that I just remember the pattern: split/operation/merge. ddply is the right function if I'm splitting a data.frame based on some columns and building a new data.frame on the results of an operation on the pieces. – Harlan Aug 19 '10 at 17:02

writing the outcome of a nested loop to a vector object in R

3 Answers3