基于R语言data.frame已有列创建哑变量d的技术需求

没问题，我来帮你搞定这个变量创建的需求！首先先明确你的原始数据集和要求：

现有数据集：example <- data.frame(a = 1:10, b = 10:1, c = 1:5)（注：因为c=1:5只有5个值，R会自动循环填充到10行，和你描述的结构一致）
需求：创建新变量d，当每行的a、b、c中至少有一个值为1、2或3时，d取1；否则取0，预期结果为d <- c(1, 1, 1, 0, 0, 1, 1, 1, 1, 1)

下面给你几种简洁高效的实现方法：

方法一：Base R 向量运算（高效简洁） #

利用rowSums和逻辑判断的特性（TRUE等价于1，FALSE等价于0），一行代码搞定：

# 先定义原始数据集
example <- data.frame(a = 1:10, b = 10:1, c = 1:5)

# 创建变量d
example$d <- as.integer(rowSums(sapply(example[, c("a", "b", "c")], function(x) x %in% 1:3)) > 0)

代码解释： #

1. sapply(example[, c("a", "b", "c")], function(x) x %in% 1:3)：对a、b、c每一列，检查每个元素是否属于{1,2,3}，返回一个10行3列的逻辑矩阵；
2. rowSums(...)：计算每行中TRUE的数量（即满足条件的变量个数）；
3. > 0：判断该行是否至少有一个满足条件的变量，返回逻辑值；
4. as.integer(...)：把逻辑值转成整数（TRUE→1，FALSE→0）。

方法二：Base R 逐行判断（直观易懂） #

用apply按行处理，逻辑更直观：

example <- data.frame(a = 1:10, b = 10:1, c = 1:5)

example$d <- apply(example[, c("a", "b", "c")], 1, function(x) any(x %in% 1:3))
example$d <- as.integer(example$d)

代码解释： #

1. apply(..., 1, function(x) any(x %in% 1:3))：按行（1表示行）遍历数据，any(...)判断该行是否有元素属于{1,2,3}，返回逻辑向量；
2. as.integer(...)：将逻辑向量转为整数向量，得到最终的d。

方法三：Tidyverse/dplyr 风格（简洁优雅） #

如果你习惯用tidyverse系列工具，dplyr的if_any函数可以让代码更简洁：

library(dplyr)

example <- data.frame(a = 1:10, b = 10:1, c = 1:5) %>%
  mutate(d = as.integer(if_any(c(a, b, c), ~ .x %in% 1:3)))

代码解释： #

1. if_any(c(a, b, c), ~ .x %in% 1:3)：检查a、b、c中任意一列是否满足元素属于{1,2,3}的条件，返回逻辑向量；
2. as.integer(...)：将逻辑向量转为整数，得到d。

验证结果 #

运行任意一种方法后，查看example$d都会得到预期结果：

example$d
# 输出：1 1 1 0 0 1 1 1 1 1

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