时间戳

首页 > 解决方案 > R：使用“for”循环和“case_when”检查多个变量。

r - R：使用“for”循环和“case_when”检查多个变量。

问题描述

我有一个数据集DT如下：
类别：数字1-9
xxx，yyy，zzz：二进制（0,1）

category xxx yyy zzz
8         1   0   0
1         0   0   0
4         0   1   1
9         0   0   1
8         0   1   0

我想使用“for”循环和“case_when”检查多个条件。
所以，我希望数据显示在最后

category xxx yyy zzz result_xxx result_yyy result_zzz
8         1   0   0       8           0          0
1         0   0   0       0           0          0
4         0   1   1       0           4          4 
9         0   0   1       0           0          9 
8         0   1   0       0           8          0

为此，我在下面编写了代码：

condition.vars <- c("xxx", "yyy", "zzz")

for(i in condition.vars){
  browser()
  DT <- DT[, condition:= case_when(
    ([[i]] == 1 & category ==1) ~ 1,
    ([[i]] == 1 & category ==2) ~ 2,
    ([[i]] == 1 & category ==3) ~ 3,
    ([[i]] == 1 & category ==4) ~ 4,
    ([[i]] == 1 & category ==5) ~ 5,
    ([[i]] == 1 & category ==6) ~ 6,
    ([[i]] == 1 & category ==7) ~ 7,
    ([[i]] == 1 & category ==8) ~ 8,
    ([[i]] == 1 & category ==9) ~ 9,
    TRUE ~ 0
  )]
  setnames(DT, "condition", paste0("result", i))
   
}

如您所料，它不起作用。
你能帮我更正我的代码吗？

标签： rfor-loopcase-when

解决方案

您不需要for循环或case_when. 如果您有数据框，则可以将其简化为：

condition.vars <- c("xxx", "yyy", "zzz")
DT[paste0('result_', condition.vars)] <- DT$category * DT[condition.vars]

#  category xxx yyy zzz result_xxx result_yyy result_zzz
#1        8   1   0   0          8          0          0
#2        1   0   0   0          0          0          0
#3        4   0   1   1          0          4          4
#4        9   0   0   1          0          0          9
#5        8   0   1   0          0          8          0

如果DT是data.table你可以这样做：

library(data.table)
DT[, paste0('result_', condition.vars):= category * .SD,.SDcols = condition.vars]

推荐阅读