Question

我正在尝试在数据框的列子集中使用mutate_at。但是，当我使用的函数是ifelse时，它似乎不是矢量化。

例如，下面的代码应该提供与原始输入相同的输出。

set.seed(1)
stack_names <- c('a','c','d','e')
stack_df <- data.frame(a = 1:10, b = 11:20, c = 21:30, d = rep(1, 10), e = rnorm(10))

#     a  b  c d          e
# 1   1 11 21 1 -0.6264538
# 2   2 12 22 1  0.1836433
# 3   3 13 23 1 -0.8356286
# 4   4 14 24 1  1.5952808
# 5   5 15 25 1  0.3295078
# 6   6 16 26 1 -0.8204684
# 7   7 17 27 1  0.4874291
# 8   8 18 28 1  0.7383247
# 9   9 19 29 1  0.5757814
# 10 10 20 30 1 -0.3053884

stack_df %>% mutate_at(.vars = vars(one_of(stack_names)), 
                         funs(ifelse(length(unique(.)) == 1, ., .)))
#    a  b  c d          e
# 1  1 11 21 1 -0.6264538
# 2  1 12 21 1 -0.6264538
# 3  1 13 21 1 -0.6264538
# 4  1 14 21 1 -0.6264538
# 5  1 15 21 1 -0.6264538
# 6  1 16 21 1 -0.6264538
# 7  1 17 21 1 -0.6264538
# 8  1 18 21 1 -0.6264538
# 9  1 19 21 1 -0.6264538
# 10 1 20 21 1 -0.6264538

这是ifelse的已知行为吗？当我尝试类似的代码，但使用不同的功能，它工作正常。

stack_df %>% mutate_at(.vars = vars(one_of(stack_names)), 
                           funs(. + mean(.)))
#       a  b    c d          e
# 1   6.5 11 46.5 2 -0.4942510
# 2   7.5 12 47.5 2  0.3158461
# 3   8.5 13 48.5 2 -0.7034258
# 4   9.5 14 49.5 2  1.7274836
# 5  10.5 15 50.5 2  0.4617106
# 6  11.5 16 51.5 2 -0.6882656
# 7  12.5 17 52.5 2  0.6196318
# 8  13.5 18 53.5 2  0.8705275
# 9  14.5 19 54.5 2  0.7079841
# 10 15.5 20 55.5 2 -0.1731856

Answer 1

ifelse的第一个参数是长度为1的逻辑向量。您只需要获得一个值。参见

ifelse(TRUE, 5:1, 1:5)
# [1] 5    # one 1 value returned, not 5

当您测试length(unique(.)) == 1时，只会获得一个TRUE或FALSE，具体取决于列中的所有值是否相同。

ifelse的所有参数应该是相同的长度。目前还不清楚你到底在做什么，所以解决方案可能并不明显。看起来您可能只想有条件地执行mutate或使用mutate_if或某种类型。也许你只想要一个普通的if陈述。

myfun <- function(x) {
    if (length(unique(x)) == 1) x else x
}
stack_df %>% mutate_at(.vars = vars(one_of(stack_names)), funs(myfun))

dplyr mutate_at和ifelse（）没有矢量化

1 个答案: