如何在未知数组维度时移除其最后一维的指定元素？

针对未知维度的数组，要移除最后一维中指定索引的元素，这里提供一个简洁高效的方案，无需分支判断或生成大量中间副本：

核心实现代码 #

toremove <- c(2,5,6)
# 示例高维数组（可替换为任意2维及以上数组）
A <- array(1:24, dim = c(2,2,2,3))

# 获取数组维度数
ndim <- length(dim(A))
# 构造索引列表：前ndim-1个维度取全部元素，最后一维排除指定索引
idx <- vector("list", ndim)
idx[-ndim] <- rep(list(quote(expr = )), ndim - 1)
idx[[ndim]] <- -toremove

# 执行子集提取，drop=F保证维度不丢失
Anew <- do.call("[", c(list(A), idx, drop = FALSE))

方案说明 #

这个方法的本质是通过构造索引列表，用do.call调用数组的提取函数[，完全模拟手动写A[, , -toremove, drop=F]的逻辑：

• quote(expr = )对应手动索引中的空逗号（即取该维度所有元素）
• 最后一维传入-toremove排除指定元素
• 加入drop=FALSE确保即使最后一维只剩一个元素，也保留原数组的维度结构

对比已有思路的优势 #

• 对比思路1：无需展开数组为向量计算绝对索引，避免了大数组下的内存开销和索引计算错误
• 对比思路2：不用针对每个维度写分支判断，代码简洁且支持任意高维数组
• 对比思路3：解决了空参数传递的问题，用quote(expr = )模拟空逗号，完美适配do.call的调用逻辑

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