在R中循环运行函数

时间:2015-06-26 21:11:15

标签: r

我需要在循环中使用以下函数,因为我有100个变量。

binning <- function (df,vars,by=0.1,eout=TRUE,verbose=FALSE) {
for (col in vars) {
  breaks <- numeric(0)
  if(eout) {
    x <- boxplot(df[,col][!df[[col]] %in% boxplot.stats(df[[col]])$out],plot=FALSE)    
    non_outliers <- df[,col][df[[col]] <= x$stats[5] & df[[col]] >= x$stats[1]]
    if (!(min(df[[col]])==min(non_outliers))) {
      breaks <- c(breaks, min(df[[col]]))
    }
  }
breaks <- c(breaks, quantile(if(eout) non_outliers else df[[col]], probs=seq(0,1, by=by)))  
  if(eout) {
    if (!(max(df[[col]])==max(non_outliers))) {
      breaks <- c(breaks, max(df[[col]]))
    }    
  }

  return (cut(df[[col]],breaks=breaks,include.lowest=TRUE))
}}

它创建一个带分档分数的变量。变量的命名约定是“原始名称”加上“_bin”。

data$credit_amount_bin <- iv.binning.simple(data,"credit_amount",eout=FALSE)

我希望该函数针对所有NUMERIC变量运行,并将转换后的bin变量存储在不同的数据框中,并使用“原始名称_bin”命名。

任何帮助都将受到高度赞赏。

1 个答案:

答案 0 :(得分:2)

使用您的功能,您可以通过lapply,循环显示所有数字值。

# some data
dat0 <- data.frame(a=letters[1:10], x=rnorm(10), y=rnorm(10), z=rnorm(10))

# find all numeric by names
vars <- colnames(dat0)[which(sapply(dat0,is.numeric))]

# target data set
dat1 <- as.data.frame( lapply(vars, function(x) binning(dat0,x,eout=FALSE)) )
colnames(dat1) <- paste(vars, "_bin", sep="")

就个人而言,我更喜欢使用向量输入而不是数据框加变量名来实现此功能。它也可以更高效地运行。