R语言-编写自定义函数

太过爱你忘了你带给我的痛 2022-06-12 14:49 255阅读 0赞

**一、函数构造器**  
每一个R函数都包括三个部分：函数名，程序主体以及参数集合，在编写自定义R函数时，需要将三个部分各自储存在一个R对象中。这里需要使用function函数，形如：  
`my_function<-function(){}`  
function函数可以将大括号内的代码内容构建成一个函数，例如：

roll<-function(){
      dice<-sample(1:6,2,replace = T)#可重复抽样
      sum(dice)
    }
    roll()

函数将返回最后一行的运行输出结果，如果最后一行不输出结果，整个函数也将不会有返回值

--------------------

**二、含参函数的构造**  
可以通过在括号里填入参数名称加以实现，具体如下：

roll2<-function(bones){
      dice<-sample(bones,2,replace = T)#可重复抽样
      sum(dice)
    }
    > roll2()
    Error in sample(bones, 2, replace = T) : 
      argument "bones" is missing, with no default
    > roll2(1:10)
    [1] 11

可以发现如果不写参数，R将报错，为了防止类似的错误发生，可以预先设置一个初始默认值给bones参数，只需在括号里用等号将默认值与参数相连即可。，形如：

roll3<-function(bones=1:6){
      dice<-sample(bones,2,replace = T)#可重复抽样
      sum(dice)
    }
    > roll3()
    [1] 9

--------------------

**三、提取函数**  
就像其他语言的IDE一样，RStudio也提供了抽取函数的功能，在R脚本中选中对应的代码段，在菜单栏中选择Code>Extract Function即可。RStudio会询问给函数取什么名字，然后用function将所需函数包装起来，并会自动检查代码段中未定义的变量，将其设置为该函数的参数。  
提取函数之后，同志们记得检查一下，主要是看看有没有什么需要设置默认值的参数，或者函数有没有返回值之类的问题。

**四、查看函数**  
不仅可以通过help查看函数解释，还可以通过直接输入函数名查看函数代码  
这就是开源的好处哦~