R中使用函数类变量做形参

Take a function as an argument

R语言,数据分组 和关于if for问题

一个求助引出的冷艳代码

有一个dataset

dataset <- c(3,4,5,6,7,2,3,4,5,6,6,1)

写一个function计算这个数据组每五个数一组的average和range,如果最后一组少于5个数的话写一个note,就是

the average and range of(3,4,5,6,7)

the average and range of(2,3,4,5,6)

the average and range of(6,1)

and a message that in the last group only 2 values

高贵冷艳不知道怎么运行的版本

cos上写代码以冷艳简短著称的月珥给出了示范代码:

dataset <- c(3,4,5,6,7,2,3,4,5,6,6,1)
MyFun=function(num,fun=c(mean,range))
{
  if(length(num)%%5!=0)
    warning('The last group has less than 5 elements!')
  tapply(num,as.integer(seq(1,by=0.2,along.with=num)),fun)
}
lapply(c(mean,range),MyFun,num=1:13)

作者附带的解释:用function类变量做形参的写法对新人来说大概太过丧心病狂了点。但是计算思路个人觉得还是可以模仿一下的:先建立一个步进1/5=0.2的数列,然后强制转换成整数型,就可以快速得到一个每5个元素一组的factor,然后再tapply就行。

单一功能版

dataset <- c(3,4,5,6,7,2,3,4,5,6,6,1)
MyFun=function(data)
{
  if(length(data)%%5!=0)
    warning('The last group has less than 5 elements!')
  tapply(data,as.integer(seq(1,by=0.2,along.with=data)),mean)
}
MyFun(dataset)
dataset

月珥:当然这样就只能用来计算平均值了。要计算range的话把mean改成range即可。

屌丝版

dataset <- c(3,4,5,6,7,2,3,4,5,6,6,1)
myfun <- function(data){
  i=1
  k=4
  data0=NULL
  data1=NULL
  data2=NULL

data2=c(mean="",min="",max="")

j <- c(seq(1,length(data),by=5)) for(i in j){ data0 <-data[i:(i+k)] if(length(data)<(i+k)){ print(sprintf('The last group has only %i values!',length(data)%%5)) data0 <-na.omit(data0) } data1 <- c(mean(data0),range(data0)) data2 <- rbind(data2,data1) i=i+k+1 } colnames(data2)=c("mean","min","max") rownames(data2)=seq(1,nrow(data2)) return(data2) } myfun(dataset)

发表评论

电子邮件地址不会被公开。 必填项已用*标注