Calculate first difference by group in R

Question

I was wondering if someone could help me calculate the first difference of a score by group. I know it should be a simple process but for some reason I'm having trouble doing it..... yikes

Here's an example data frame:

score <- c(10,30,14,20,6)

group <- c(rep(1001,2),rep(1005,3))

df <- data.frame(score,group)

> df 
  score group
1    10  1001
2    30  1001
3    14  1005
4    20  1005
5     6  1005

And here's the output I was looking for.

Thanks in advance.

user20650 · Accepted Answer · 2014-04-15T18:43:12.030

14

This is one way using base R

df$diff <- unlist(by(df$score , list(df$group) , function(i) c(NA,diff(i))))

or

df$diff <- ave(df$score , df$group , FUN=function(i) c(NA,diff(i)))

or using data.table - this will be more efficient for larger data.frames

library(data.table)
dt <- data.table(df)
setkey(dt,group)
dt[,diff:=c(NA,diff(score)),by=group]

edited Apr 15 '14 at 18:43

answered Apr 15 '14 at 18:25

user20650

24,654
5
56
91

Thanks, this one worked great. I used the base R syntax but need to install data.table. – Richard Apr 15 '14 at 19:27

Miha Trošt · Answer 2 · 2015-02-18T08:39:18.557

9

Another approach using dplyr:

library(dplyr)

score <- c(10,30,14,20,6)
group <- c(rep(1001,2),rep(1005,3))
df <- data.frame(score,group)

df %>%
  group_by(group) %>%
  mutate(first_diff = score - lag(score))

edited Feb 18 '15 at 08:39

answered Apr 16 '14 at 08:07

Miha Trošt

2,002
22
25

score 1 · Answer 3 · answered Apr 15 '14 at 18:02

1

Although not exactly what you are looking for, ddply within the 'plyr' package can be used ta calculate the differences by group

library(plyr)
out<-ddply(df,.(group),summarize,d1=diff(score,1))

answered Apr 15 '14 at 18:02

Steve Reno

1,304
3
14
21

Thank you Steve, I tried plyr but it was running really slow on my computer. – Richard Apr 15 '14 at 19:27

score 0 · Answer 4 · answered Apr 15 '14 at 18:15

This should do the trick, although it uses loops rather than an apply function, so there is likely room for improvement in code clarity/efficiency

out = numeric()
#out[1] will always be NA
out[1] = NA
for(i in 2:nrow(df)){
  if(df$group[i]==df$group[(i-1)]){
    out[i]=df$score[i]-df$score[(i-1)]
  } 
  else {
    out[i]=NA
  }
}
out
[1]  NA  20  NA   6 -14

Calculate first difference by group in R

4 Answers4

Linked

Related