根据条件向一列添加值

Question

我想根据两列更新一列

我的示例数据框包含 3 列

df <- data.frame(n1 = c(1,2,1,2,5,6),
                 n2 = c("a", "a", "a", NA, "b", "c"),
                 n3 = c("red", "red", NA, NA, NA, NA))

df

  n1   n2   n3
1  1    a  red
2  2    a  red
3  1    a <NA>
4  2 <NA> <NA>
5  5    b <NA>
6  6    c <NA>

我想将red 名称添加到行号3 和4，条件是如果n1（即1,2）的值与n2（即a）匹配，即使第四行（n1 不匹配 n2）。

重点是如果n2 == a，并且n1的值与a相关联，那么n3的值与n1的值在同一行应该加上red.

我想要的输出

  n1   n2   n3
1  1    a  red
2  2    a  red
3  1    a  red
4  2 <NA>  red
5  5    b <NA>
6  6    c <NA>

对这个案例有什么建议吗？我希望我的解释足够清楚。由于我的数据很长，我正在努力寻找一个好的来处理它。

Answer 1

在base R 中，根据'n1' 的unique 值（其中'n2' 是“a”）创建一个逻辑向量以子集'df' 的行，然后分配'n3' 对应于具有来自 'n3' 的第一个非 NA 元素的元素

i1 <- with(df, n1 %in% unique(n1[n2 %in% 'a']))
df$n3[i1] <- na.omit(df$n3[i1])[1]

-输出

> df
  n1   n2   n3
1  1    a  red
2  2    a  red
3  1    a  red
4  2 <NA>  red
5  5    b <NA>
6  6    c <NA>

Answer 2

更新：

df %>% 
  mutate(group = rep(row_number(), each=2, length.out = n())) %>% 
  group_by(group) %>% 
  mutate(n3 = ifelse(n1 %in% c(1,2) & any(n2 %in% "a", na.rm = TRUE), "red", n3)) %>% 
  ungroup() %>% 
  select(-group)

我们可以使用ifelse 语句和使用any 定义的条件。

library(dplyr)
df %>% 
  mutate(n3 = ifelse(n1==1 | n1==2 & any(n2[3:4] %in% "a"), "red", n3))

  n1   n2   n3
1  1    a  red
2  2    a  red
3  1    a  red
4  2 <NA>  red
5  5    b <NA>
6  6    c <NA>

Answer 3

library(dplyr)
library(tidyr)

df %>% 
  group_by(n1) %>%
  fill(n3) %>%
  group_by(n2) %>%
  fill(n3)
# # A tibble: 6 × 3
# # Groups:   n2 [4]
#      n1 n2    n3   
#   <dbl> <chr> <chr>
# 1     1 a     red  
# 2     2 a     red  
# 3     1 a     red  
# 4     2 NA    red  
# 5     5 b     NA   
# 6     6 c     NA