编写一个接受整数向量并返回逻辑向量的函数

Question

我偶然发现了一个我正在努力解决的问题。

"编写一个函数，它接受一个整数向量并返回一个逻辑向量，当输入为偶数时为 TRUE，当输入为奇数时为 FALSE，当输入为非有限时返回 NA（即如果它是 Inf，-Inf , NA, NaN)。检查它是否适用于正、负、零和非有限输入。"

f <- function(x,y)
{
  if (x==Inf)
  {
    print ("NA")
  }else 
    if(x%%2=0)
{print ("TRUE")
    }else
        {print("FALSE")}
  if (y==Inf)
    {print("NA")
    }
  else
    if (y%%2==0)
  {print ("TRUE")
  } else
  {print ("FALSE")
  }
  print(c(x+y, x-y, x^4-y^3, x^2+y^2, y*x))
}

通过我的眼睛，我看不出我哪里出错了。我收到一个错误代码说“错误：“}”中的意外'}'

任何指导将不胜感激。

Answer 1

你可以稍微缩短你的函数并写

f <- function(x) { x %% 2 == 0 }

测试

f(c(1:6, Inf, -Inf, NA, NaN))
# [1] FALSE  TRUE FALSE  TRUE FALSE  TRUE    NA    NA    NA    NA

Answer 2

您当前方法的问题是您正在尝试对向量进行操作，但您的 if - else 逻辑实际上是对标量（即单个值）进行操作。我建议使用矢量化解决方案，例如 ifelse() 函数。我们可以将一堆对ifelse 的调用链接在一起，但在这里使用dplyr 包中的case_when 可能会更容易：

library(dplyr)

f <- function(x) {
    result <- case_when(
        x %% 2 == 0 ~ TRUE,
        x %% 2 == 1 ~ FALSE,
        is.infinite(x) ~ NA,
        is.nan(x) ~ NA,
        TRUE ~ NA
    )

    return(result)
}

编辑：

我更喜欢@markus 的答案，这可能正是您的老师所期望的。但是像我上面提到的那样使用case_when 会使您期望的映射变得明确。所以，从代码清晰的角度来看，发生了什么是完全清楚的。