如何在R中创建函数将因子所有水平替换为原水平的最后3个字符

可用解决方案 #

第一步：复用你已编写的finales函数 #

你的基础函数逻辑没有问题，可直接复用：

finales <- function(v, n){
  substr = substr(v, nchar(v) - (n - 1), nchar(v))
  return(substr)
}

第二步：编写end.levels函数 #

end.levels <- function(f) {
  # 提取原因子的水平，操作不会影响原始传入的因子
  ori_levels <- levels(f)
  # 调用finales函数生成新水平（取每个水平最后3位）
  new_levels <- finales(ori_levels, 3)
  # 创建副本修改水平，避免修改原数据
  res_f <- f
  levels(res_f) <- new_levels
  return(res_f)
}

测试验证 #

用你提供的测试因子运行：

fac = factor(c("Alto","Bajo", "Mediano", "Bajo", "Alto", "Alto", "Mediano"))
renom <- end.levels(fac)
# 执行测试用例
all(renom == c("lto", "ajo", "ano", "ajo", "lto", "lto", "ano"))

运行后将返回TRUE，符合预期要求。

原有代码的错误原因 #

1. 括号位置错误：你写的nchar(labels(fac) - 3)是先对字符类型的水平值执行减3操作，字符和数值不能直接运算，因此触发非数值参数报错，正确写法应为nchar(labels(f)) - 3。
2. 变量名不匹配：函数传入参数为f，但代码中直接调用了全局变量fac，函数无法通用处理其他传入的因子。
3. 操作对象错误：你直接对因子元素做截取，而不是修改因子的levels属性，不符合仅修改水平、不修改原始数据的要求。
4. 不存在rename直接修改因子水平的原生用法，R中修改因子水平直接对levels(因子对象)赋值即可。

内容的提问来源于stack exchange，提问作者Abel Pérez