创建循环函数,用于计算R中其他列的一列的值

时间:2011-10-13 12:28:17

标签: r function loops

对于数据框AB:

AB<-data.frame(ID=c(1,2,4),A=c(2,8,8),B=c(6,2,2),dE=c(0,0,0))

我想应用以下公式:AB$dE=AB$B/AB$A

ID A  B  dE
1  2  6  0
2  8  2  0
4  8  2  0

将上述内容转换为:

ID A  B  dE
1  2  6  3
2  8  2  0.25
4  8  2  4

因为我有几个文件包含A和B的不同列名,所以编写一个函数会更实用,比如

dEs <- function(data,nume,denom){ 
        #define which datafile and numerator/denom column
        #so in case of AB this becomes AB$dE; 
        # i dont know the correct way to do this.
        dE=data.'$dE'
        start=data.'$'.nume #to become AB$A
        end=data.'$'.denom #to become AB$B
        for(i in data){
          dE[i] <- (start[i]/end[i])
        }
}

这样我就可以在必要时改变分子/分母。

1 个答案:

答案 0 :(得分:2)

你不需要循环,因为R是矢量化的:

> AB <- data.frame(ID=c(1,2,4),A=c(2,8,8),B=c(6,2,2),dE=c(0,0,0))
> AB <- transform(AB, dE=B/A)
> AB
  ID A B   dE
1  1 2 6 3.00
2  2 8 2 0.25
3  4 8 2 0.25

如果你真的想要一个函数,你可以使用[[]]来选择列,因为data.frame只是一个列表(假设nume和denom是带有你想要的列名称的字符向量): / p>

dEs <- function(data,nume,denom){ 
  data$dE <- data[[nume]] / data[[denom]]
}