行跨行的有条件突变（按组/ID）？

Question

我有一个很大的数据集，我希望提供一些帮助。下面给出了一个示例：

   id  id_row  material
1  1     1     1
2  1     2     1
3  1     3     1
4  2     1     1
5  2     2     2
6  2     3     1
7  3     1     1
8  3     2     1
9  3     3     1
10 4     1     1
11 4     2     2 

我想根据相同ID（跨行）的材料中的材料值添加一个新列。在新的COLUM中，我希望所有具有材料中值1和2的ID（跨行）（例如，值为99），如果不存在，则返回1或2。 类似的事情：

   id  id_row  material  new_column
1  1     1     1             1
2  1     2     1             1
3  1     3     1             1
4  2     1     1             99
5  2     2     2             99
6  2     3     1             99
7  3     1     2             2
8  3     2     2             2
9  3     3     2             2
10 4     1     1             99
11 4     2     2             99

我一直在网上寻找解决方案，没有任何运气，并且尝试使用dplyr和group_by，mutate和ifelse没有任何运气。先感谢您！

Answer 1

尝试以下方法：

library(tidyverse)

tribble(
  ~id, ~id_row, ~material,
  1, 1, 1,
  1, 2, 1,
  1, 3, 1,
  2, 1, 1,
  2, 2, 2,
  2, 3, 1,
  3, 1, 2,
  3, 2, 2,
  3, 3, 2,
  4, 1, 1,
  4, 2, 2
) |> 
  group_by(id) |> 
  mutate(new_column = if_else(any(material == 2) & any(material == 1), 99, NA_real_),
         new_column = if_else(is.na(new_column), material, new_column))
#> # A tibble: 11 × 4
#> # Groups:   id [4]
#>       id id_row material new_column
#>    <dbl>  <dbl>    <dbl>      <dbl>
#>  1     1      1        1          1
#>  2     1      2        1          1
#>  3     1      3        1          1
#>  4     2      1        1         99
#>  5     2      2        2         99
#>  6     2      3        1         99
#>  7     3      1        2          2
#>  8     3      2        2          2
#>  9     3      3        2          2
#> 10     4      1        1         99
#> 11     4      2        2         99

^{在2022-05-25创建 reprex package （v2.0.1）< /sup>}