R数据.可以使用变量名在分组中进行分配

Question

我正在尝试使用一个字符变量作为新列的名称以及by来汇总data.table。

library(data.table)
dt <- data.table(g = rep(1:3, 4), xa = runif(12), xb = runif(12))
# desired output
dt[, .(sa = mean(xa)), by = g]
g       sa
1: 1 1.902360
2: 2 2.149041
3: 3 2.586044

问题是以下代码仍然返回整个数据表，而没有减少到g的唯一值。

cn <- paste0('s', 'a')
# returns all rows
dt[, (cn) := mean(xa), by = g][]
g        xa         xb        sa
1: 1 0.3423699 0.81447505 0.4755900
2: 2 0.0932055 0.06853225 0.5372602
3: 3 0.2486223 0.13286546 0.6465111
4: 1 0.6942175 0.66405944 0.4755900
5: 2 0.7225208 0.83110248 0.5372602
6: 3 0.9898293 0.09520907 0.6465111
7: 1 0.3523753 0.72743182 0.4755900
8: 2 0.5504942 0.01966303 0.5372602
9: 3 0.3523625 0.55257436 0.6465111
10: 1 0.5133974 0.39650089 0.4755900
11: 2 0.7828203 0.89909528 0.5372602
12: 3 0.9952302 0.16872205 0.6465111

如何获得常用的汇总数据表？(这是一个简化的例子。在我的实际问题中，会有多个名称传递到一个循环中(

Answer 1

有一个挂起的PR，这将使此类操作更加容易，data.表#4304.一旦在当前设计中实现，查询将看起来像：

dt[, .(cn = mean(xa)), by = g, env = list(cn="sa")]
#       g        sa
#   <int>     <num>
#1:     1 0.2060352
#2:     2 0.1707827
#3:     3 0.6850591

PR分支的安装

remotes::install_github("Rdatatable/data.table@programming")

数据

library(data.table)
dt <- data.table(g = rep(1:3, 4), xa = runif(12), xb = runif(12))

Answer 2

使用包裹在list(.(mean(xa))(列周围的setNames或

dt[, setNames(.(mean(xa)), cn), by = g]
#  g        sa
#1: 1 0.2010599
#2: 2 0.4710056
#3: 3 0.4871248

---

或获得汇总输出后的setnames

setnames(dt[, mean(xa), by = g], 'V1', cn)[]

---

在data.table中，:=运算符用于在原始数据集中创建/修改列。但是，当在tidyverse上下文中使用时，此运算符不同

library(dplyr)
dt %>%
group_by(g) %>% 
summarise(!! cn := mean(xa), .groups = 'drop')
# A tibble: 3 x 2
#      g    sa
#  <int> <dbl>
#1     1 0.201
#2     2 0.471
#3     3 0.487