为何switch与case_when在R语言管道操作中无法正常运行？

报错原因 #

1. switch管道写法报错原因 #

switch()的第一个参数要求是长度为1的向量（比如单个字符/数值），用来做分支判断。但你用管道的时候，前一步生成的矩阵被自动传到了switch()的第一个参数位置（也就是.的位置），原本应该作为判断依据的param变成了第二个参数，完全搞反了switch的参数顺序。这就导致switch试图把矩阵当成分支判断的依据，而矩阵是长度远大于1的对象，直接触发EXPR must be a length 1 vector错误。

2. case_when写法报错原因 #

case_when()要求每个分支都是条件~结果的双向公式，而且你犯了两个错误：

• 没把管道传来的矩阵（.）传给rowMeans2和rowSds，函数找不到要计算的对象
• 管道把矩阵直接传给了case_when的第一个参数，但case_when的第一个参数应该是一系列双向公式，不是矩阵，所以触发Case 1 (.) must be a two-sided formula错误。

解决方案 #

要在管道里实现分支逻辑，得把switch或case_when用大括号或者匿名函数包裹，让管道传来的对象只在需要的地方使用，不干扰分支函数的参数顺序：

方法1：修正switch的管道写法 #

用大括号包裹switch，这样管道传来的矩阵只会在rowMeans2/rowSds里作为参数使用，switch的第一个参数还是param：

myFunc <- function(i, column, B, param) {
  est <- unlist(column) %>%
    sample(i, replace = TRUE) %>%
    matrix(nrow = B, ncol = i) %>%
    {
      switch(
        param,
        "mean" = rowMeans2(., na.rm = TRUE),
        "sd" = rowSds(., na.rm = TRUE),
        stop("Invalid `param` value. Valid values are 'mean' or 'sd.'")
      )
    } %>%
    mean()
  
  print(est)
}

方法2：用case_when实现管道分支 #

同样用大括号包裹，确保每个分支都正确接收管道传来的矩阵，同时符合case_when的语法：

myFunc <- function(i, column, B, param) {
  est <- unlist(column) %>%
    sample(i, replace = TRUE) %>%
    matrix(nrow = B, ncol = i) %>%
    {
      case_when(
        param == "mean" ~ rowMeans2(., na.rm = TRUE),
        param == "sd" ~ rowSds(., na.rm = TRUE),
        TRUE ~ stop("Invalid `param` value. Valid values are 'mean' or 'sd.'")
      )
    } %>%
    mean()
  
  print(est)
}

测试验证 #

运行以下代码可以验证修正后的函数正常工作：

# 示例数据
set.seed(123)
column <- data.frame(X1 = rnorm(100))

# 参数设置
i <- 5
B <- 100
param <- "mean"

# 加载依赖包
library(matrixStats)

# 调用函数
myFunc(i, column, B, param)

内容的提问来源于stack exchange，提问作者CBRF23