r - 根据条件添加数字

Question

my_df <- tibble(
  b1 = c(2, 6, 3, 6, 4, 2, 1, 9, NA), 
  b2 = c(NA, 4, 6, 2, 6, 6, 1, 1, 7), 
  b3 = c(5, 9, 8, NA, 2, 3, 9, 5, NA), 
  b4 = c(NA, 6, NA, 10, 12, 8, 3, 6, 2),
  b5 = c(2, 12, 1, 7, 8, 5, 5, 6, NA),
  b6 = c(9, 2, 4, 6, 7, 6, 6, 7, 9),
  b7 = c(1, 3, 7, 7, 4, 2, 2, 9, 5),
  b8 = c(NA, 8, 4, 5, 1, 4, 1, 3, 6),
  b9 = c(4, 5, 7, 9, 5, 1, 1, 2, NA),
  b10 = c(14, 2, 4, 2, 1, 1, 1, 1, 5))

大家好，

希望你们一切都好。我有一个这样的 df(非常大的一个)，如果 b6、67、b8 或 b9 中有 2，我想告诉 R 将 b1 中的值加 10。再次感谢您的期待。

Answer 1

我们可以在 case_when 中创建一个逻辑条件，方法是对列子集 b6:b9 的行总和来查找该行是否在任何行中至少有 2，然后将 10 添加到 b1 或否则返回原始列

library(dplyr)
my_df <- my_df %>% 
    mutate(b1 = case_when(rowSums(select(cur_data(), b6:b9) == 2, 
        na.rm = TRUE) > 0 ~ b1 + 10, TRUE ~ b1))

-输出

my_df
# A tibble: 9 x 10
     b1    b2    b3    b4    b5    b6    b7    b8    b9   b10
  <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl>
1     2    NA     5    NA     2     9     1    NA     4    14
2    16     4     9     6    12     2     3     8     5     2
3     3     6     8    NA     1     4     7     4     7     4
4     6     2    NA    10     7     6     7     5     9     2
5     4     6     2    12     8     7     4     1     5     1
6    12     6     3     8     5     6     2     4     1     1
7    11     1     9     3     5     6     2     1     1     1
8    19     1     5     6     6     7     9     3     2     1
9    NA     7    NA     2    NA     9     5     6    NA     5

---

或者也可以使用if_any

my_df %>% 
    mutate(b1 = case_when(if_any(b6:b9, `%in%`, 2) ~ b1 + 10, TRUE ~ b1))

-输出

# A tibble: 9 x 10
     b1    b2    b3    b4    b5    b6    b7    b8    b9   b10
  <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl>
1     2    NA     5    NA     2     9     1    NA     4    14
2    16     4     9     6    12     2     3     8     5     2
3     3     6     8    NA     1     4     7     4     7     4
4     6     2    NA    10     7     6     7     5     9     2
5     4     6     2    12     8     7     4     1     5     1
6    12     6     3     8     5     6     2     4     1     1
7    11     1     9     3     5     6     2     1     1     1
8    19     1     5     6     6     7     9     3     2     1
9    NA     7    NA     2    NA     9     5     6    NA     5

---

或者在 base R 中相同

i1 <- rowSums(my_df[6:9] == 2, na.rm = TRUE) > 0
my_df$b1[i1] <- my_df$b1[i1] + 10

---

或者使用 Reduce/lapply 和 %in%

i1 <- Reduce(`|`, lapply(my_df[6:9], `%in%`, 2))
my_df$b1[i1] <- my_df$b1[i1] + 10