Question

我有一些数据集，其中一列是“曝光”，一列是“事件数”，还有几列标记了各种类别类型。

 Exposure<-c(10,2.1,2.8,4.5,21)
 NoEvents <- c(1,0,2,0,0)
 Cat1<-as.factor(c("A","A","B","A","B"))
 Cat2<-as.factor(c("X","Y","Y","Y","X"))
 Cat3<-as.factor(c("u","v","u","w","w"))
 dataTest<-data.frame(Exposure,NoEvents,Cat1,Cat2,Cat3)
 dataTest


    Exposure     NoEvents        Cat1         Cat2        Cat3
    10.0         1               A            X            u
    2.1          0               A            Y            v
    2.8          2               B            Y            u
    4.5          0               A            Y            w
    21.0         0               B            X            w

我现在想（灵活）计算和绘制针对所选类别类型（Cat1，Cat2或Cat3）的合计频率（NoEvents / Exposure）。对于固定的类别列，例如Cat1，我可以定义以下函数

freq_Cat <- function(data,Cat1){
  data_aggr<-aggregate(. ~ Cat1, data[,c("Exposure","NoEvents","Cat1")], sum)
  data_aggr[,"frequency"] <- data_aggr$NoEvents/data_aggr$Exposure
  return(data_aggr)
}

然后用

 ggplot(freq_Cat(dataTest,Cat1), aes(x=Cat1,y=frequency)) + 
 geom_bar(stat="identity",fill="dodgerblue",col="black")

我希望使函数freq_Cat以及绘图更加灵活，以便我可以灵活地选择按哪种类别类型/列（Cat1，Cat2或Cat3）进行聚合，而无需复制粘贴并将Cat1替换为另一个一个。

Answer 1

如果将类别作为数字传递，则可以计算要使用的变量。

freq_Cat <- function(data,Cat){
  Var  = paste("Cat", Cat, sep="")
  Form = formula(paste(". ~", Var))
  data_aggr<-aggregate(Form, data[,c("Exposure","NoEvents",Var)], sum)
  data_aggr[,"frequency"] <- data_aggr$NoEvents/data_aggr$Exposure
  return(data_aggr)
}

但是现在不用像使用

那样调用它了

freq_Cat(dataTest, Cat=1)

或者，如果您想通过名称引用类别变量，则可以使用：

freq_Cat <- function(data,Cat){
  Form = formula(paste(". ~", Cat))
  data_aggr<-aggregate(Form, data[,c("Exposure","NoEvents",Cat)], sum)
  data_aggr[,"frequency"] <- data_aggr$NoEvents/data_aggr$Exposure
  return(data_aggr)
}

然后访问如下功能：

freq_Cat(dataTest, Cat="Country")

R：根据所选类别标准绘制汇总频率

1 个答案: