R中如何不转换为数据框 直接通过变量名按类别值筛选table对象

报错原因 #

base R的subset函数未针对table类设计适配方法，默认实现无法识别table的维度名作为查询变量，因此会抛出object 'Age' not found的错误。

实现思路 #

table本质是带维度名属性的数组，我们可以通过dimnames属性定位目标维度的位置，再通过数组索引提取对应子集，全程不需要转换为数据框，完全基于base R实现。

示例代码（泰坦尼克数据集场景） #

# 加载数据集
data("Titanic")
# 定位Age维度的索引
age_dim_idx <- which(names(dimnames(Titanic)) == "Age")
# 定位Adult在Age维度中的位置
adult_val_idx <- which(dimnames(Titanic)[[age_dim_idx]] == "Adult")
# 构造全量维度索引，非筛选维度保留全部值
index_list <- rep(list(quote(expr = )), length(dim(Titanic)))
index_list[[age_dim_idx]] <- adult_val_idx
# 提取结果，drop=TRUE自动删除仅存单一取值的Age维度
titanic_adult <- do.call(`[`, c(list(Titanic), index_list, drop = TRUE))

运行后得到的titanic_adult和你提供的预期输出结构完全一致。

通用封装函数 #

你可以直接使用下面的通用函数，支持任意table对象按维度名筛选：

# 输入：x为待筛选table对象，var为目标维度名，value为目标维度取值
subset_table <- function(x, var, value) {
  if (!inherits(x, "table")) stop("输入参数x必须为table类型")
  dim_names <- names(dimnames(x))
  if (!var %in% dim_names) stop(paste0("维度", var, "不存在于输入table中"))
  target_dim <- which(dim_names == var)
  target_val_pos <- which(dimnames(x)[[target_dim]] == value)
  if (length(target_val_pos) == 0) stop(paste0("取值", value, "不存在于维度", var, "中"))
  index_list <- rep(list(quote(expr = )), length(dim(x)))
  index_list[[target_dim]] <- target_val_pos
  do.call(`[`, c(list(x), index_list, drop = TRUE))
}

# 调用示例
subset_table(Titanic, "Age", "Adult")

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