r - 如何计算有条件的平均值？

Question

这个问题已经有答案了:

Calculating mean when 2 conditions need met in R (2 个回答)

已关闭 8 年前。

下面是生成可重现数据帧的脚本:

id <- c(1:20)
a <- as.numeric(round(runif(20,-40,40),2))
b <- as.numeric(round(a*1.4+60,2))
df <- as.data.frame(cbind(id, a, b))

我想计算“a”在不同条件下“b”的平均值。例如，当 -40 =< a < 0 时，“b”的平均值是多少；当 0=< a <=40 时，“b”的平均值是多少。

非常感谢!

Answer 1

这是一个快速的 data.table 解决方案(假设 coef 是 a)

library(data.table)
setDT(df)[, .(MeanASmall = mean(b[-40 <= a & a < 0]),
              MeanABig = mean(b[0 <= a & a <= 40]))]
#    MeanASmall MeanABig
# 1:   33.96727    89.46

如果范围有限，您也可以使用基本 R 快速完成此操作

sapply(split(df, df$a >= 0), function(x) mean(x$b))
#     FALSE     TRUE 
#  33.96727 89.46000