计算等式

Question

的累积和假设在R I有以下的数据帧dat：

a <- c(6,6,6,7,9,10,11,13,16,17,19,20,22,10,23,25,32,32,34,35,6) 
b <- c(1,1,0,1,0,1,0,1,1,0,0,0,1,0,1,0,0,0,0,0,1) 
q <- c(3,2,4,2,4,2,2,3,0,5,0,6,1,0,3,4,9,1,2,0,12) 

dat <- data.frame(a, b, q) 
dat <- dat[order(dat$a),] 

对于a每个级别，我必须计算f = with(dat, sum(q[a==that level]*b[a==that level])/sum(b[a==that level]))。例如，对于​​,f=(3*1+2*1+4*0+12*1)/(3+2+4+12)=17/21。

所以我写的代码如下：

dat$qb <- dat$q * dat$b 

v1 <- as.numeric(by(dat$qb, dat$a, FUN=sum)) 
dat$sqb <- rep(v1, table(dat$a)) 

v2<-as.numeric(by(dat$q, dat$a, FUN=sum)) 
dat$sumq <- rep(v2, table(dat$a)) 

dat$f <- dat$sqb/dat$sumq 

现在我只需要列a和f：

dat <- unique(dat[,c(1,7)]) 

我的下一步是计算公式f = with(dat, sum(q[a==that level]*b[a==that level])/sum(b[a==that level]))的累积和：

dat <- replace(dat, is.na(dat), 0) 
dat$F <- cumsum(dat$f) 

最终，我ne版的F为其a<=18最大值：

Ft <- max(dat$F[dat$a<=18]) 

这是一个模拟的一部分，并且该部分的目标是找到的​​值。我认为有更快的方法来编写这些代码块。

• 任何更快的R代码块计算Ft将不胜感激。

Answer 1

我们可以通过“A”使用tidyverse到组，用“Q”乘以“B”计算“F”，获得由“Q”的sum的sum和鸿沟。用0.1N

library(tidyverse) 
dat1 <- dat %>% 
      group_by(a) %>% 
      summarise(f = sum(b*q)/sum(q)) %>% 
      mutate(F = cumsum(replace(f, is.na(f), 0))) 

过滤replace荷兰国际集团NA之后创建“F”为“F”的cumsum其中“a”是小于或等于18的行中，提取“F”的列，并获得max

dat1 %>% 
    filter(a <= 18) %>% 
    .$F %>% 
    max 
#[1] 3.809524 

---

或者另一种选择是data.table

library(data.table) 
setDT(dat)[, .(f= sum(b*q)/sum(q)) , a][is.na(f), f := 0 
      ][, F := cumsum(f)][a <= 18, max(F)] 
#[1] 3.809524