Advertisement
Guest User

Untitled

a guest
Jun 16th, 2019
104
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.02 KB | None | 0 0
  1. day_prob <-
  2. list(
  3. A = c(0.06, 0.06, pmax(dchisq(1:13, 3)*1, 0.06))
  4. , B = c(0.06, 0.06, pmax(dchisq(seq(1, 25, 2), 5)*6, 0.06))
  5. )
  6.  
  7. day_prob %>%
  8. lapply(function(x){
  9. tibble(
  10. Day = 1:length(x)
  11. , Prob = x
  12. )
  13. }) %>%
  14. bind_rows(.id = "Group") %>%
  15. ggplot(aes(x = Day
  16. , y = Prob
  17. , col = Group)) +
  18. geom_line() +
  19. geom_point() +
  20. scale_color_brewer(palette = "Dark2")
  21.  
  22. make_obs <- function(group, day){
  23. sapply(1:length(group), function(idx){
  24. rbinom(1, 1, day_prob[[group[idx]]][day[idx]] )
  25. })
  26. }
  27.  
  28. set.seed(12345)
  29. example_data <-
  30. tibble(
  31. Group = rep(c("A", "B"), each = 50)
  32. , Ind = 1:100
  33. , Day = 1
  34. ) %>%
  35. complete(nesting(Group, Ind), Day = 1:15) %>%
  36. mutate(
  37. Obs = make_obs(Group, Day)
  38. )
  39.  
  40. example_data %>%
  41. group_by(Group, Day) %>%
  42. summarise(Prop_pos = mean(Obs)) %>%
  43. ungroup() %>%
  44. ggplot(aes(x = Day
  45. , y = Prop_pos
  46. , col = Group)) +
  47. geom_line() +
  48. scale_color_brewer(palette = "Dark2")
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement