R语言for循环中如何以列值为名称动态新建逻辑列

原代码问题说明 #

你写的循环存在4个核心错误，导致无法正常运行：

• 赋值语法错误：df$i是固定访问/创建列名为i的列，不会解析循环变量i存储的实际值，动态列名赋值需要使用双括号[[ ]]语法
• grepl参数顺序写反：函数正确参数顺序是grepl(匹配模式, 待匹配向量)，你把待匹配列和匹配值的位置颠倒了
• 未过滤唯一值中的NA：直接用unique(df$b)会把NA也纳入循环，生成无意义的NA列
• 语法不完整：for循环缺少闭合的大括号}，且未做精确匹配锚定，遇到存在包含关系的字符串会出现误匹配。另外你给出的预期结果中第4行b列标注为A属于笔误，和测试数据中第4行b为C的设定不一致。

修正后的循环实现 #

直接运行以下代码即可得到目标结果：

# 构建测试数据框
df <- data.frame('a'=c(1,2,3,4,5), 'b'=c("A",NA,"B","C","A"))
# 提取b列非NA的唯一匹配值
match_list <- unique(na.omit(df$b))

# 循环动态生成逻辑列
for (i in match_list) {
  # 加^和$做精确匹配，避免字符串包含关系导致的误判
  # 用[[i]]动态解析变量值作为列名
  df[[i]] <- grepl(paste0("^", i, "$"), df$b)
}

# 将所有NA位置的逻辑值统一替换为FALSE
df[is.na(df)] <- FALSE

运行后打印df即可得到正确输出：

a    b     A     B     C
1 1    A  TRUE FALSE FALSE
2 2 <NA> FALSE FALSE FALSE
3 3    B FALSE  TRUE FALSE
4 4    C FALSE FALSE  TRUE
5 5    A  TRUE FALSE FALSE

更简洁的tidyverse实现（可选） #

如果不想写循环，可以用tidyr的宽表转换函数一行实现，自动处理NA和列生成逻辑：

library(dplyr)
library(tidyr)

df <- df %>%
  mutate(flag = TRUE) %>%
  pivot_wider(names_from = b, values_from = flag, values_fill = FALSE)

运行结果和循环写法完全一致。

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