3

我想将tapply结果作为新列添加到原始数据框中。

这是我的数据框：

 dat <- read.table(text = " category birds    wolfs     snakes
                   yes        3        9         7
                   no         3        8         4
                   no         1        2         8
                   yes        1        2         3
                   yes        1        8         3
                   no         6        1         2
                   yes        6        7         1
                   no         6        1         5
                   yes        5        9         7
                   no         3        8         7
                   no         4        2         7
                   notsure    1        2         3
                   notsure    7        6         3
                   no         6        1         1
                   notsure    6        3         9
                   no         6        1         1   ",header = TRUE)

我想将每个类别的平均值作为一列添加到数据框中。我使用：tapply(dat$birds, dat$category, mean)获取每个类别的平均值，但我没有找到将其添加到数据集中的方法，以至于在新列中我将获得相关类别的平均值。

4

3 回答 3

6

你可以使用ave从base

  dat$mbirds <- with(dat, ave(birds, category, FUN=mean))

如果你想使用tapply

  mbirds1 <- with(dat, tapply(birds, category, mean))
  dat$mbirds1 <- mbirds1[match(dat$category,names(mbirds1))]

  head(dat)
  #  category birds wolfs snakes mbirds mbirds1
 #1      yes     3     9      7  3.200   3.200
 #2       no     3     8      4  4.375   4.375
 #3       no     1     2      8  4.375   4.375
 #4      yes     1     2      3  3.200   3.200
 #5      yes     1     8      3  3.200   3.200
 #6       no     6     1      2  4.375   4.375

或者你可以使用data.table哪个会很快

 library(data.table)
 setDT(dat)[,mbirds1:= mean(birds), by=category]
于 2014-09-01T11:29:07.280 回答
3

这是一个aggregate答案。在它的参数中使用一个公式使它变得简单而简单。

> a <- aggregate(birds~category, dat, mean)
> cb <- cbind(dat, mean = a[,2][match(dat[[1]], a[,1])])
> head(cb)
#  category birds wolfs snakes  mean
#1      yes     3     9      7 3.200
#2       no     3     8      4 4.375
#3       no     1     2      8 4.375
#4      yes     1     2      3 3.200
#5      yes     1     8      3 3.200
#6       no     6     1      2 4.375
于 2014-09-01T12:12:02.480 回答
2

您可以使用这样的dplyr包轻松实现

dat <- dat %>% group_by(category) %>% mutate(mbirds=mean(birds))

关于 dplyr 包的更多信息可以在这里找到。

您可以在 akrun 的回答中找到其他软件包的方法。

于 2014-09-01T11:21:14.333 回答