R - 使用函数过滤向量

Question

我有一个与此类似的功能:

isGoodNumber <- function(X) 
{
if (X==5) return(TRUE) else return(FALSE)
}

I have a vector:
v<-c(1,2,3,4,5,5,5,5)

我想获得一个包含 v 元素的新向量，其中 isGoodNumber(v) == TRUE

我该怎么做？

尝试了 v [ isGoodNumber(v) == TRUE ] 但不起作用:-)

谢谢!!

Answer 1

有一个名为“Filter ”的函数可以完全满足您的要求:

Filter( isGoodNumber, v)
#[1] 5 5 5 5

可以选择创建一个向量化的函数，可以通过使用 Vectorize 函数(已说明)或使用 ifelse 编写(也可以提到)，并且可以选择类似“过滤器”的功能

 isGoodNumber3 <- function(X) 
   { X[ ifelse(X==5, TRUE,FALSE)]
   }

 isGoodNumber3(v)
#[1] 5 5 5 5