使用循环在 R 中定义 case_when 的值

Question

我目前正在使用case_when 在我的数据中定义一个新变量：

data[,46] <- NA

data[,46] <- case_when(
   data[,35] ==  1 ~ data[,36],
   data[,35] ==  2 ~ data[,37],
   data[,35] ==  3 ~ data[,38],
   data[,35] ==  4 ~ data[,39],
   data[,35] ==  5 ~ data[,40],
   data[,35] ==  6 ~ data[,41],
   data[,35] ==  7 ~ data[,42],
   data[,35] ==  8 ~ data[,43],
   data[,35] ==  9 ~ data[,44],
   data[,35] ==  10 ~ data[,45]
)

我正在尝试编写一个循环来提高此功能的效率，但遇到了一些麻烦。这是我尝试过的：

for (j in 1:10) {
data[,46] <- case_when(
   data[,35] ==  j ~ data[,35+j]
)
}

但是，这将为我的所有 data[,46] 值返回 NA。对可能出了什么问题有任何想法吗？如有必要，我很乐意提供示例数据，但我认为这与我犯了一个简单的编程错误更相关。提前致谢！

Answer 1

您所要做的就是记住 R 是矢量化的。
您将data[, 35] 与整数1 到10 进行比较，并将data[, 35 + <1 to 10>] 分配回data[, 35]。所以你要做的就是

data[, 35] <- data[, 35 + data[, 35]]

如果data[, 35] 中的值不在1:10 中，那么ifelse 将更合适。

data[, 35] <- ifelse(data[, 35] %in% 1:10, data[, 35 + data[, 35]], data[, 35])

Answer 2

您可能需要[j] 如下所示将其迭代存储在data[,46] 中

for (j in 1:10) {
data[,46][j]<- case_when(
   data[,35] ==  j ~ data[,35+j]
)}