R语言中mapply的vapply等效函数是什么？空输入问题如何解决？

核心问题 #

基础R中，mapply作为sapply的多变量版本，当输入参数长度为0时会返回list()，但实际期望得到符合函数返回类型的空对象（如integer(0)或指定维度的空数组）。是否存在类似vapply的安全变体，或是推荐的处理方式？

解决方案 #

1. 使用purrr::pmap系列函数（推荐） #

purrr包的pmap族函数支持明确指定返回类型，完美适配mapply的多变量场景，同时解决空输入的类型问题：

示例1：返回整数类型

library(purrr)

size_of_union <- function(A, B) length(union(A, B))

# 正常场景
x <- list(1:3, 2, 3)
y <- list(3, 2, numeric(0)) 
pmap_int(list(x, y), size_of_union)
#> [1] 3 1 1

# 边缘场景
x <- integer(0)
y <- integer(0)
pmap_int(list(x, y), size_of_union)
#> integer(0)

示例2：返回矩阵类型

对于返回固定长度向量的函数（如range返回长度为2的向量），可先收集结果再转换为矩阵：

range_of_intersect <- function(A, B) range(intersect(A, B))

# 正常场景
x <- list(1:3, 2, 3)
y <- list(3, 2, numeric(0)) 
result <- pmap(list(x, y), range_of_intersect)
do.call(cbind, result)
#>      [,1] [,2] [,3]
#> [1,]    3    2  Inf
#> [2,]    3    2 -Inf

# 边缘场景
x <- numeric(0)
y <- numeric(0)
result <- pmap(list(x, y), range_of_intersect)
if (length(result) == 0) {
  matrix(numeric(0), nrow = 2, ncol = 0)
} else {
  do.call(cbind, result)
}
#>      [,1]
#> [1,]   
#> [2,]   

2. 自定义基础R版vmapply #

若不想依赖第三方包，可手动实现类似vapply的多变量安全函数：

vmapply <- function(FUN, ..., FUN.VALUE, USE.NAMES = TRUE) {
  args <- list(...)
  input_length <- if (length(args) == 0) 0 else max(lengths(args))
  
  if (input_length == 0) {
    # 处理空输入，返回指定类型的空对象
    if (is.matrix(FUN.VALUE)) {
      matrix(FUN.VALUE[0], nrow = nrow(FUN.VALUE), ncol = 0)
    } else {
      FUN.VALUE[0]
    }
  } else {
    # 正常场景调用mapply并验证类型
    res <- mapply(FUN, ..., USE.NAMES = USE.NAMES)
    stopifnot(all(vapply(res, function(x) identical(typeof(x), typeof(FUN.VALUE)), logical(1))))
    
    if (is.matrix(FUN.VALUE)) {
      do.call(cbind, res)
    } else {
      unlist(res)
    }
  }
}

# 测试示例1
size_of_union <- function(A, B) length(union(A, B))
vmapply(size_of_union, x = integer(0), y = integer(0), FUN.VALUE = integer(1))
#> integer(0)

# 测试示例2
range_of_intersect <- function(A, B) range(intersect(A, B))
vmapply(range_of_intersect, x = numeric(0), y = numeric(0), FUN.VALUE = numeric(2))
#>      [,1]
#> [1,]   
#> [2,]   

3. 临时手动处理边缘情况 #

如果仅需临时解决空输入问题，可在mapply调用后直接检查并转换返回值：

# 示例1处理
res <- mapply(size_of_union, x = integer(0), y = integer(0))
if (identical(res, list())) res <- integer(0)

# 示例2处理
res <- mapply(range_of_intersect, x = numeric(0), y = numeric(0))
if (identical(res, list())) res <- matrix(numeric(0), nrow = 2, ncol = 0)

总结 #

• 优先选择purrr::pmap_*系列函数，语法简洁且类型安全；
• 依赖基础R时，自定义vmapply可统一处理类型校验和空输入场景；
• 临时场景下，手动检查并转换空返回值是快速解决方案。

内容的提问来源于stack exchange，提问作者jan-glx