如何创建R自定义函数匹配编码与主列表并自动生成分类统计列

云上老周

2026-9-25

完全可以通过自定义函数实现，不需要逐行编写重复的ifelse语句。

前置处理注意事项

首先需要确保你的主分类列表code_list是命名列表，列表的元素名就是最终生成的分类列名，示例如下：

code_list <- list(
  category_1 = c("101", "102"),
  category_2 = c("201", "202", "203")
)
# 如果不想修改原有code_list的定义，也可以手动批量命名
# names(code_list) <- paste0("category_", seq_along(code_list))

自定义函数实现

以下函数完整覆盖你提出的三项需求：

library(dplyr)
library(purrr)

match_code_category <- function(df, code_list, 
                                code_col = "code", 
                                value_col = "expend",
                                # 二分类校验相关可选参数
                                check_binary = FALSE,
                                binary_col = "bin",
                                binary_positive = "1") {
  # 1. 批量生成基础分类列
  category_cols <- imap_dfc(code_list, function(codes, col_name) {
    ifelse(df[[code_col]] %in% codes, df[[value_col]], 0)
  })
  res <- bind_cols(df, category_cols)
  
  # 2. 按需生成二分类校验列
  if (check_binary) {
    valid_cols <- imap_dfc(code_list, function(codes, col_name) {
      ifelse(df[[code_col]] %in% codes & df[[binary_col]] == binary_positive, df[[value_col]], 0)
    }) %>% rename_with(~paste0(.x, "_valid"))
    res <- bind_cols(res, valid_cols)
  }
  return(res)
}

调用示例

# 仅生成基础分类列，对应需求1、2
result <- match_code_category(table, code_list)

# 同时生成校验列，对应需求3
result_with_check <- match_code_category(table, code_list, check_binary = TRUE)

调用后输出结果和你给出的期望格式完全一致。