我正在寻找一种操作数据框内多个变量的有效方法。现在我正在使用dplyr,但是这对于更多的变量来说很麻烦。 假设我有以下数据框,其中brd是汽车品牌,ye是一年,type是汽车类型,cy和hp是类型特征。在一个循环中总结并生成多个变量
brd <-c("BMW","BMW","BMW","Volvo","Volvo", "Volvo","BMW","BMW","BMW","Volvo","Volvo","Volvo")
ye <- c(99,99,99,99,99,99,98,98,98,98,98,98)
type <- c(1,2,3,1,2,3,1,2,3,1,2,3)
cy <- c(1895,1991,1587,2435,2435,1596,1991,1588,1984,1596,1991,1588)
hp <- c(77,110,80,103,103,75,110,77,93,75,110,77)
df <- as.data.frame(brd)
df$ye <- ye
df$type <- type
df$cy <- cy
df$hp <- hp
df
brd ye type cy hp
1 BMW 99 1 1895 77
2 BMW 99 2 1991 110
3 BMW 99 3 1587 80
4 Volvo 99 1 2435 103
5 Volvo 99 2 2435 103
6 Volvo 99 3 1596 75
7 BMW 98 1 1991 110
8 BMW 98 2 1588 77
9 BMW 98 3 1984 93
10 Volvo 98 1 1596 75
11 Volvo 98 2 1991 110
12 Volvo 98 3 1588 77
每一年,我想计算的产品特性的总和同一品牌的其他所有产品,并将其添加为新的变量数据框。现在,我使用dplyr这样的:
library(dplyr)
df <- df %>% group_by(brd, ye) %>%
mutate(sumall_cy = sum(cy),
sumall_hp = sum(hp))
df <- df %>%
mutate(sumother_cy = sumall_cy-cy,
sumother_hp = sumall_li-hp)
所以,我得到
brd ye type cy hp sumall_cy sumall_hp sumother_cy sumother_hp
<fctr> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl>
1 BMW 99 1 1895 77 5473 267 3578 190
2 BMW 99 2 1991 110 5473 267 3482 157
3 BMW 99 3 1587 80 5473 267 3886 187
4 Volvo 99 1 2435 103 6466 281 4031 178
5 Volvo 99 2 2435 103 6466 281 4031 178
6 Volvo 99 3 1596 75 6466 281 4870 206
7 BMW 98 1 1991 110 5563 280 3572 170
8 BMW 98 2 1588 77 5563 280 3975 203
9 BMW 98 3 1984 93 5563 280 3579 187
10 Volvo 98 1 1596 75 5175 262 3579 187
11 Volvo 98 2 1991 110 5175 262 3184 152
12 Volvo 98 3 1588 77 5175 262 3587 185
有没有更有效的方法?我正在考虑像这个stata代码循环:
foreach x of varlist hp cy {
bysort ye: egen sumall_`x'= sum(`x')
gen sumother_`x'=(sumall_`x' -`x')}
任何意见将不胜感激。
您可以在以'df'开头的管道中调用'left_join'(例如'left_join(。,summarize())'来生成组级摘要,然后像你一样进行mutate'这里。它不会做更少的工作,但所有的工作都在同一个管道内,如果你关心的话。 – ulfelder