对特定变量值的条件操作

Question

我正在寻找一种通过变量值对数据框中的变量编写条件操作的方法，但我无法完全理解如何去做。

假设我有以下动物收容所摄入量的每日数据：

shelters <- data.frame(
    date = rep(seq(as.Date("2014-02-01"), length=10, by = "1 day"),10),
    animal = sample(c("dog","cat","goldfish"), 100, replace = T),
    intake = sample(1:10, 100, replace = T))

我想设置一个规则来增加数据框，以便当“动物”变量中某个动物的摄入量总和达到给定值时，在达到给定日期之后的每个“摄入量”值变为0。

假设每只动物的最大值如下：

dog = 90
cat = 100
goldfish = 85

我倾向于使用 cumsum，但如何在 cumsum 公式中指定特定的动物值？

Answer 1

试试：

 shelters$indx <- setNames(c(90,100,85), c("dog", "cat", "goldfish") )[as.character(shelters$animal)]

 library(dplyr)
 shelters%>% 
 arrange(date, animal)%>%
 group_by(animal) %>% 
 mutate(Sum=cumsum(intake), intake=ifelse(Sum >indx, 0, intake)) %>% 
 select(-indx, -Sum) %>%
 head()
 #       date animal intake
 #1 2014-02-01    cat      1
 #2 2014-02-01    cat      4
 #3 2014-02-01    cat      2
 #4 2014-02-01    cat      2
 #5 2014-02-01    dog      5
 #6 2014-02-01    dog      9

更新

如果你想让总和与最大值相同

  res <- shelters %>%
  arrange(date, animal)%>% 
  group_by(animal) %>% 
  mutate(Sum=cumsum(intake), index2 = ifelse(Sum>indx,intake-(Sum-indx), intake),
            intake= ifelse(index2<0,0, index2))%>% 
  select(-indx, -Sum, -index2)

  res %>% summarize(Sum=sum(intake))
  #Source: local data frame [3 x 2]

  #    animal Sum
  #1      cat 100
  #2      dog  90
  #3 goldfish  85