条件突变以在 R 中使用 dplyr 分配新颜色

Question

我有一个看起来像这样的数据框

value <- c(0,0.1,0.2,0.4,0,0.05,0.05,0.5)
time <- c(0,0,0,0,1,1,1,1)
ID <- c(1,2,3,4,1,2,3,4)

test <- data.frame(value, time, ID)

 value time ID
1  0.00    0  1
2  0.10    0  2
3  0.20    0  3
4  0.40    0  4
5  0.00    1  1
6  0.05    1  2
7  0.05    1  3
8  0.50    1  4

我想当时间点 '0' 的值大于 0.1 时分配颜色“红色” 到所有 ID 值，当值小于 0.1 时分配颜色“蓝色” 我希望我的 data.frame 看起来像这样

 value time ID  color
1  0.00    0  1 blue
2  0.10    0  2 blue 
3  0.20    0  3 red
4  0.40    0  4 red
5  0.00    1  1 blue
6  0.05    1  2 blue
7  0.05    1  3 red
8  0.50    1  4 red

这是我迄今为止尝试过的，但我没有得到所有时间点的颜色

test %>% 
  group_by(ID) %>% 
  mutate(color = case_when(value > 0.1 & time ==0 ~ "red", 
                           value < 0.1 & time ==0 ~ "blue"))

非常感谢任何提示

更新::编辑

---

我看到的一种方法是添加

test %>% 
  group_by(ID) %>% 
  mutate(color = case_when(value > 0.1 & time ==0 ~ "red", 
                           value < 0.1 & time ==0 ~ "blue"))%>% 
 tidyr:fill(color, .diretction ="updown") 

还有其他建议吗？

Answer 1

你试过 if_else() 吗？

test %>% 
  mutate(color = if_else(value > 0.1 & time ==0,"red","blue"))

或者，嵌套 if_else() 更进一步：

test %>% mutate(color = if_else(value > 0.1 & time ==0,
                               "red",
                               if_else(time == 0,
                                       "blue",
                                       "0")
                               )
               )

如果规则对整个 ID 组是通用的，您可以使用组的 max()（或 min()）吗？

test %>%
  group_by(ID) %>% 
  mutate(color = if_else(max(value) > 0.1 & time ==0,
                               "red",
                               if_else(time == 0,
                                       "blue",
                                       "not 0")
                               )
               )

将 ID == 1 的数据更改为高于 0.1 和低于 0.1 的值使这一点更加明显：

library(dplyr)

value <- c(2,0.1,0.2,0.4,0,0.05,0.05,0.5)
time <- c(0,0,0,0,0,1,1,1)
ID <- c(1,2,3,4,1,2,3,4)

test <- data.frame(value, time, ID)

test %>%
  group_by(ID) %>% 
  mutate(color = if_else(min(value) > 0.1 & time ==0,
                         "red",
                         if_else(time == 0,
                                 "blue",
                                 "not 0")
  )
  )

  value  time    ID color
  <dbl> <dbl> <dbl> <chr>
1  2        0     1 blue 
2  0.1      0     2 blue 
3  0.2      0     3 blue 
4  0.4      0     4 red  
5  0        0     1 blue 
6  0.05     1     2 not 0
7  0.05     1     3 not 0
8  0.5      1     4 not 0

Answer 2

  group_by(ID) %>% 
  mutate(color = case_when(value > 0.1 & time ==0 ~ "red", 
                           value < 0.1 & time ==0 ~ "blue"))%>% 
 tidyr:fill(color, .diretction ="updown")```