Question

我想根据类别对数据集应用函数。给出以下数据框

pet <- c(rep("cat",5),rep("dog",5))
year <- c(rep(1991:1995,2))
karma <- c(5,4,1,1,1,6,4,3,2,6)
df <- data.frame(pet,year,karma)

看起来像这样

   pet year karma
1  cat 1991     5
2  cat 1992     4
3  cat 1993     1
4  cat 1994     1
5  cat 1995     1
6  dog 1991     6
7  dog 1992     4
8  dog 1993     3
9  dog 1994     2
10 dog 1995     6

我想在每年的业力列上执行操作。如果我想应用像 sum 这样的函数，可以使用ddply来完成：

ddply(df, .(year),summarize, sum(karma))

我如何将它应用于我自己编写的函数，例如

calc <- function(d,c){(d*5+c*7)/12}

其中 d 是与每个给定年份的狗的业力相对应的值，而 c 对应于猫的业力。

理想情况下，我希望在此数据框中附加五个条目，其中两者，一年和上面函数计算的业力值。这样做的最佳方式是什么？

（非常抱歉，如果这是微不足道的，但这次我真的找不到类似的问题。）

Answer 1

您可以使用spread扩展数据框，然后使用mutate来实现您的功能

library('tidyr')
library('dplyr')
df %>% 
 spread(pet, karma, drop = FALSE) %>% 
 mutate(karma = calc(dog, cat), pet = "both") %>% 
 select(year, pet, karma) %>%
 rbind(df)

按组

1 个答案: