R中有各大语言都有的if/else,也有自己特有ifelse。
单值条件判断
R中的if类似其他语言(如python),只能接受单个的TRUE或FALSE,不能接受NA,否则会报错。
> if(NA) message('NA not allowed')
Error in if (NA) message("NA not allowed") :
missing value where TRUE/FALSE needed
if 和else配合时的整体语法如下:
if (单个逻辑值) {
执行语句
}else if() { # 这里的else必须和前一个if的 } 在同一行
执行语句2
}
另外还有赋值:
> x <- if(rnorm(1) > 0) 'here' else 'there'
> x
[1] "here"
逻辑值向量判断
R中特有的向量化条件判断ifelse。参数中的test,yes和no都能够接受向量输入,但最好长度一致,否则会出现循环或省略,输出难以预料的结果。
ifelse(test, yes, no)
> ifelse(rbinom(10,1,.45),'head','tail')
[1] "head" "head" "tail" "tail" "head" "tail" "head" "head" "tail" "head"
> ifelse(rbinom(46,1,.5),letters,LETTERS)
[1] "A" "b" "c" "D" "E" "f" "G" "h" "i" "J" "K" "L" "m" "N" "O" "p" "Q" "R" "s" "t" "u" "V" "w" "X" "y"
[26] "z" "a" "B" "C" "d" "E" "F" "g" "h" "I" "J" "k" "l" "M" "N" "O" "P" "q" "r" "s" "t"
类似于普通的if,在条件部分如果有NA的话,输出一定也是NA,因此一定要小心!!!
三分类及三分类以上
使用switch()来代替繁杂的if else堆叠。
switch(EXPR,
case = action)
EXPR参数可以接受character或number,且长度必须为1。如果是string,那么case部分可以安排一个unnamed argument,如果是numeric则不行。
> switch('default',a=pi,b=rnorm(5),c=3,pi)
[1] 3.141593 #由于‘default’匹配不上a,b,c,因此选择了unnamed的参数pi
显式循环
类似于python,也是while(),for(),以及R特有的repeat()的搭配。
R的repeat相当于python的while() action,需要break或next(相当于python的continue?)来控制长度。相比与直接使用while,repeat的特点是action部分的内容至少会被执行一次。
R中for的使用方法与python很相似,不再赘述。
注意!R中的匹配使用 %in% ,返回逻辑向量!而match()则返回配对值的下标。
> x= sample(1:10,5,replace = T)
> x
[1] 4 6 3 7 3
> x %in% c(1,3,7)
[1] FALSE FALSE TRUE TRUE TRUE
> match(x,c(1,3,7))
[1] NA NA 2 3 2