R语言编写函数移除列指定因子水平失败，求代码修正

你的代码没达到预期的核心原因有两个：

1. 将因子水平设为NA只是修改了该水平的标签，并没有真正移除这个水平，它依然会保留在因子的水平列表中；
2. purrr::map的使用方式不对，直接在函数内修改全局数据框时，需要明确的赋值操作，且原代码的参数传递逻辑不够清晰。

修正后的代码方案 #

方案一：用purrr包实现（贴合你的原思路）

iris_df <- iris
iris_df$Species2 <- iris_df$Species

# 遍历目标列，移除指定水平
purrr::walk(
  .x = c("Species", "Species2"),
  .f = function(col) {
    # 重新定义因子：保留除"setosa"外的所有水平
    iris_df[[col]] <<- factor(
      iris_df[[col]],
      levels = setdiff(levels(iris_df[[col]]), "setosa")
    )
  }
)

# 验证结果
levels(iris_df$Species)
# 输出：[1] "versicolor" "virginica"

方案二：Base R实现（更轻量）

如果不想依赖purrr，用base R的lapply也能快速完成：

iris_df <- iris
iris_df$Species2 <- iris_df$Species

cols_to_edit <- c("Species", "Species2")
iris_df[cols_to_edit] <- lapply(
  iris_df[cols_to_edit],
  function(col) factor(col, levels = setdiff(levels(col), "setosa"))
)

levels(iris_df$Species)

补充说明 #

• 如果你只是想把"setosa"对应的观测值转为缺失、同时保留该水平，可以用iris_df[[col]][iris_df[[col]] == "setosa"] <- NA，但这和你要移除水平的需求不同；
• setdiff(levels(col), "setosa")用于生成剔除指定水平后的新水平列表，再用factor()重新构建因子，这是移除因子水平的标准做法；
• 用purrr::walk而非map是因为我们只需要执行修改数据的副作用，不需要返回结果列表。

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