Question

我想知道是否有人可以帮我计算分组的第一个分数差异。我知道这应该是一个简单的过程，但出于某种原因我无法做到这一点......哎呀

以下是一个示例数据框：

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

这是我正在寻找的输出。

提前致谢。

Answer 1

这是使用基础R

的一种方式

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

或

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

或者使用data.table - 这对于更大的data.frames来说会更有效

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

Answer 2

使用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))

Answer 3

虽然不完全符合您的要求，但可以使用“plyr”包中的ddply ta按组计算差异

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

Answer 4

这应该可以解决问题，虽然它使用循环而不是应用函数，因此代码清晰度/效率可能有提升空间

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

在R中按组计算第一个差异

4 个答案: