r - 将一列派生数据添加到 data.frame

Question

假设我有一个简单的销售数据表

> df<-data.frame(country=c("A", "A", "B", "B"), outlet=c(1,2,1,2), sales=c(300, 900,10,40))
> df
  country outlet sales
1       A      1   300
2       A      2   900
3       B      1    10
4       B      2    40

并希望添加一列，显示每个出口在该国家/地区的所有销售额中所占的比例。我可以用split, 迭代然后重新组合 using来做到这一点rbind，但这对我来说看起来很丑陋

> do.call("rbind",lapply(split(df, df$country), function(x) { x$frac <- NA; tot<-sum(x$sales); for (o in x$outlet) {s<-x[x$outlet== o,]$sales; x[x$outlet == o,]$frac <- s/tot}; return(x)}))
    country outlet sales frac
A.1       A      1   300 0.25
A.2       A      2   900 0.75
B.3       B      1    10 0.20
B.4       B      2    40 0.80

有没有一种更简洁的方法来完成这个简单的任务（除了为它编写一个函数，只是将丑陋的东西扫到一个脚本中）？

（对于奖励积分，有没有办法防止rbind将行名称添加A.1到结果中data.frame？）

Answer 1

另一种选择：

df$frac <- df$sales / ave(df$sale, df$country, FUN = sum)
df
#  country outlet sales frac
#1       A      1   300 0.25
#2       A      2   900 0.75
#3       B      1    10 0.20
#4       B      2    40 0.80

Answer 2

这是一个更简单的方法

x <- tapply(df$sales, df$country, sum) #total sales by country
df$frac <- df$sales/x[match(df$country, names(x), nomatch=-1)] 
df

Answer 3

您可以像这样直接将新列添加到数据框中：

value <- # The code to calculate frac
df$frac <- value

我将它分成两行以使其更具可读性。

您可以deparse.level = 0在rbind调用中设置让函数构造没有标签。