R语言中如何为指示变量值为1的对应factor因子水平添加星号
R实现因子水平按指示变量追加星号的最优方案
基于你已加载的tidyverse依赖,推荐以下两种实现,按需选择:
方案1:通用易读写法(适合绝大多数场景)
核心逻辑是先转字符做条件拼接,再转回因子:
# 核心处理代码 data <- data %>% mutate( # 注意:你原数据通过cbind生成时indicator被自动转为字符类型,所以判断用"1" # 如果是数值类型的indicator,把判断条件改为 indicator == 1 即可 factors = as.factor(if_else(indicator == "1", paste0(factors, "*"), as.character(factors))) )
方案2:因子专属高效写法(适合大数据量、因子水平多的场景)
直接调用tidyverse生态中forcats包的fct_relabel()修改因子水平,无需反复转换类型:
data <- data %>% mutate( factors = fct_relabel(factors, ~if_else(indicator[match(.x, factors)] == "1", paste0(.x, "*"), .x)) )
两种方案运行后输出的结果均符合你的预期:
# A tibble: 4 × 2 factors indicator <fct> <chr> 1 a* 1 2 b 0 3 c* 1 4 d 0
内容的提问来源于stack exchange,提问作者PaulB
相关产品推荐
相关产品推荐

