R语言中使用%in%时如何应用多条件?数据框列赋值报错求助
问题分析与代码修正
原代码无法运行的核心问题有两个:
- 在dplyr管道中直接用
df[,paste0("x",1:3)]引用数据框不符合语法逻辑,且会生成一个包含所有列元素的长逻辑向量,无法与数据框的行一一对应 %in%在这里是逐元素判断,返回的结果维度和数据框行数不匹配,导致ifelse无法正确生成每行对应的结果
修正方案1(推荐dplyr写法)
用across遍历目标列,结合rowSums判断每行是否存在符合条件的值:
df <- data.frame(x1 = LETTERS[1:10], x2 = LETTERS[2:11], x3 = LETTERS[3:12]) df <- df %>% mutate(m = as.integer(rowSums(across(x1:x3, ~ .x %in% c("A", "B"))) > 0))
across(x1:x3, ~ .x %in% c("A", "B")):遍历x1到x3列,对每列的每个元素判断是否属于"A"或"B",生成一个逻辑矩阵rowSums(...) > 0:计算每行中TRUE的数量,大于0说明该行至少有一个符合条件的值as.integer():把逻辑值转成1(TRUE)或0(FALSE)
修正方案2(按行处理写法)
用rowwise让代码逐行执行判断:
df <- df %>% rowwise() %>% mutate(m = as.integer(any(c_across(x1:x3) %in% c("A", "B")))) %>% ungroup()
rowwise():设置按行处理数据c_across(x1:x3):把当前行的x1到x3列合并成一个向量any(...):判断向量中是否存在符合条件的元素ungroup():取消按行分组,避免影响后续操作
修正方案3(基础R写法)
如果不用dplyr,也可以直接用基础R实现:
df <- data.frame(x1 = LETTERS[1:10], x2 = LETTERS[2:11], x3 = LETTERS[3:12]) df$m <- as.integer(rowSums(df[, c("x1", "x2", "x3")] %in% c("A", "B")) > 0)
内容的提问来源于stack exchange,提问作者An116
相关产品推荐
相关产品推荐

