R中的组中位数答案

【问题标题】：Median by group in RR中的组中位数
【发布时间】：2020-01-04 09:04:11
【问题描述】：

我有以下数据框，如果值高于组的中位数，我想引入一个虚拟变量。

df<-data.frame(group=rep(c("A","B","c"),3), value1=c(1:9))
m<-aggregate(. ~ group, data=df, FUN=median)
names(m)[2]<-"median"
df<-merge(df,m, by="group", all.x = T)
df$median_0_1<-ifelse(df$median<df$value1,1,0)

有没有更优雅的方法来做到这一点？

而且，我可以调整它以将虚拟对象设置为高于或低于第三四分位数吗？

而且，这是一种可靠的可靠方式吗？

非常感谢。

【问题讨论】：

标签： r group-by median base

【解决方案1】：

如果您想保持数据框中的行数相同，请使用 ave

df$median_0_1 <- with(df, as.integer(value1 > ave(value1, group, FUN = median)))

这也可以通过dplyr完成

library(dplyr)
df %>% group_by(group) %>% mutate(median_0_1 = as.integer(value1 > median(value1)))

或data.table

library(data.table)
setDT(df)[, median_0_1 := as.integer(value1 > median(value1)), group]

【讨论】：

感谢您的快速回答。我不需要保留行，但无论如何这更干净，我相信更容易理解。

【解决方案2】：

优雅在旁观者的眼中，但你怎么喜欢这个。

df <- within(df, {
  median <- ave(value1, group, FUN=median)
  median_0_1 <- ifelse(median < value1, 1, 0)
  quantile3 <- ave(value1, group, FUN=function(x) quantile(x, probs=.3))
  quantile_0_1 <- ifelse(quantile3 < value1, 1, 0)
})
df
#   group value1 quantile_0_1 quantile3 median_0_1 median
# 1     A      1            0       2.8          0      4
# 2     B      2            0       3.8          0      5
# 3     c      3            0       4.8          0      6
# 4     A      4            1       2.8          0      4
# 5     B      5            1       3.8          0      5
# 6     c      6            1       4.8          0      6
# 7     A      7            1       2.8          1      4
# 8     B      8            1       3.8          1      5
# 9     c      9            1       4.8          1      6

【讨论】：