如何在 R 中跨多个具有字母数字范围的列使用 ifelse 来创建新列？

Question

我有一个包含 200 多列和 +200 万个观测值的数据集。在此数据集中，有 11 列（所有列都以“DIAG”开头）带有字母数字类别，我想创建一个新列，如果我感兴趣的类别在该特定范围之间，则该列将收到 1，否则，0.

我试过类似的方法，但它不起作用：

data$GROUP_12 <- ifelse(across(contains('DIAG') >= "A15" & <= "A19", 
                 ifelse(across(contains('DIAG') >= "A50" & <= "A64", 1, 0))))

我感兴趣的范围是：A15-A19、A50-A64、A80-B09、B15-B19、B25-B34。所以我必须对每个范围都这样做，但所有这些都被添加到 GROUP_12 列。

使用可重现的数据进行编辑：

data <- data.frame(DIAG_PRINC = c("A25", "B32", "O90"),
                   DIAG_SECUN = c("A16", "Y55", "K97"),
                   DIAGSEC1 = c("B15", "J55", "Y97"),
                   DIAGSEC2 = c("L16", "B55", "A97"),
                   DIAGSEC3 = c("W10", "S57", "T44"),
                   DIAGSEC4 = c("A64", "A16", "Y55"),
                   DIAGSEC5 = c("A80", "D33", "V12"),
                   DIAGSEC6 = c("M16", "N55", "X97"),
                   DIAGSEC7 = c("B16", "Y57", "O58"),
                   DIAGSEC8 = c("V45", "C23", "Q97"),
                   DIAGSEC9 = c("F98", "R44", "A54"))

有什么建议吗？

Answer 1

根据提供的示例和提到的逻辑，我们可以使用 if_any（如果特定行中的任何列具有该值）或 if_all（如果该列中的所有列特定行中选定的列的值）到 return TRUE/FALSE。然后用 + 或 as.integer 包装以将逻辑强制转换为二进制

library(dplyr)
vals_range <- c(sprintf("A%02d", c(15:19, 50:64, 80)),
     sprintf("B%02d", c(1:9, 15:19, 25:34)))
data <- data %>% 
  mutate(GROUP_12 = +(if_any(starts_with('DIAG'), ~ .x  %in% vals_range)))

-输出

data
DIAG_PRINC DIAG_SECUN DIAGSEC1 DIAGSEC2 DIAGSEC3 DIAGSEC4 DIAGSEC5 DIAGSEC6 DIAGSEC7 DIAGSEC8 DIAGSEC9 GROUP_12
1        A25        A16      B15      L16      W10      A64      A80      M16      B16      V45      F98        1
2        B32        Y55      J55      B55      S57      A16      D33      N55      Y57      C23      R44        1
3        O90        K97      Y97      A97      T44      Y55      V12      X97      O58      Q97      A54        1

如何在 R 中跨多个具有字母数字范围的列使用 ifelse 来创建新列？

How to use ifelse across multiple columns with alphanumeric range in R to create a new column?

r

if-statement