如何將tapply結果添加到現有數據幀

Question

我想將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)來獲得每個類別的平均值，但是我沒有找到一種方法將它添加到數據集中，以至於在新的列中我將具有相關類別的均值。

Answer 1

您可以使用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這將是快速

​​

Answer 2

您可以dplyr包實現，很容易像這樣

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

更多dplyr包的信息可以發現here。

你可以在akrun的答案中找到其他軟件包的方法。

Answer 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