请考虑以下内容:
library(tidyverse)
df <- tibble(x = rnorm(100), y = rnorm(100, 10, 2), z = x * y)
df %>%
mutate_all(funs(avg = mean(.), dev = sd(.), scaled = (. - mean(.)) / sd(.)))
有没有一种方法可以避免通过引用mean
和sd
列两次调用avg
和dev
。我的想法是类似
df %>%
mutate_all(funs(avg = mean(.), dev = sd(.), scaled = (. - avg) / dev))
显然,这是行不通的,因为没有列avg
和dev
,而是x_avg
,x_dev
,y_avg
,{{1} }等。
在y_dev
中是否有一种好方法,可以使用funs
工具以编程方式创建这些列引用,以便可以引用由rlang
的先前命名参数创建的列(当funs
是.
时,我将引用x
和x_mean
来计算x_dev
,依此类推)?
答案 0 :(得分:5)
我认为将数据转换为长格式会更容易
library(tidyverse)
set.seed(111)
df <- tibble(x = rnorm(100), y = rnorm(100, 10, 2), z = x * y)
df %>%
gather(key, value) %>%
group_by(key) %>%
mutate(avg = mean(value),
sd = sd(value),
scaled = (value - avg) / sd)
#> # A tibble: 300 x 5
#> # Groups: key [3]
#> key value avg sd scaled
#> <chr> <dbl> <dbl> <dbl> <dbl>
#> 1 x 0.235 -0.0128 1.07 0.232
#> 2 x -0.331 -0.0128 1.07 -0.297
#> 3 x -0.312 -0.0128 1.07 -0.279
#> 4 x -2.30 -0.0128 1.07 -2.14
#> 5 x -0.171 -0.0128 1.07 -0.148
#> 6 x 0.140 -0.0128 1.07 0.143
#> 7 x -1.50 -0.0128 1.07 -1.39
#> 8 x -1.01 -0.0128 1.07 -0.931
#> 9 x -0.948 -0.0128 1.07 -0.874
#> 10 x -0.494 -0.0128 1.07 -0.449
#> # ... with 290 more rows
由reprex package(v0.2.1.9000)于2018-11-04创建
答案 1 :(得分:2)
这可能对您有用:
avg <- quo(mean(.))
dev <- quo(sd(.))
res <- df %>%
mutate_all(funs(avg = !!avg, dev = !!dev, scaled = (. - !!avg) / !!dev))
确认它有效:
res0 <- df %>%
mutate_all(funs(avg = mean(.), dev = sd(.), scaled = (. - mean(.)) / sd(.)))
identical(res, res0)
# [1] TRUE
答案 2 :(得分:1)
这似乎有些令人费解,但它可以起作用:
scaled <- function(col_name, x, y) {
col_name <- deparse(substitute(col_name))
avg <- eval.parent(as.symbol(paste0(col_name, x)))
dev <- eval.parent(as.symbol(paste0(col_name, y)))
(eval.parent(as.symbol(col_name)) - avg) / dev
}
df %>%
mutate_all(funs(avg = mean(.), dev = sd(.), scaled = scaled(., "_avg", "_dev")))