R中如何基于字符变量的特定字符串快速创建新数值变量

实现方案 #

推荐用命名映射向量的方式实现，不需要嵌套多层ifelse，所有规则集中维护，代码简洁易读。

基础R实现（无需额外依赖包） #

核心思路是先提取V1的首字符，再通过预先定义的字符-数值映射表直接匹配赋值：

# 1. 预先定义所有首字符到目标值的映射规则，所有调整只需要修改这部分
value_map <- c(
  "A" = 3,
  "B" = 12,
  "0" = 44,
  "1" = 6,
  "2" = 27
  # 剩余0-9、A-Z的对应规则直接按上述格式追加即可
)

# 2. 提取首字符并匹配赋值
my.df$value <- value_map[substr(my.df$V1, 1, 1)]

# 若需要设置未匹配到的字符的默认值，可补充如下代码，数值按需求修改
# my.df$value[is.na(my.df$value)] = 0

运行上述代码得到的结果和你嵌套ifelse的输出完全一致，数据量越大，运行效率优势越明显。

tidyverse风格实现（适合dplyr用户） #

如果日常使用tidyverse生态，也可以用case_when函数实现，逻辑同样比嵌套ifelse清晰：

library(dplyr)
library(stringr)

my.df <- my.df %>%
  mutate(
    value = case_when(
      str_sub(V1, 1, 1) == "A" ~ 3,
      str_sub(V1, 1, 1) == "B" ~ 12,
      str_sub(V1, 1, 1) == "0" ~ 44,
      str_sub(V1, 1, 1) == "1" ~ 6,
      str_sub(V1, 1, 1) == "2" ~ 27,
      # 其余规则直接向下追加即可
      TRUE ~ 27 # 末行设置未匹配场景的默认值
    )
  )

两种方案都可以避免写几十行嵌套ifelse的冗余代码，规则变更时只需要修改映射部分即可，维护成本极低。

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