R中的参数检查

Question

在 R 中，我写了一个类似的函数：

fun <- function(A, B, C, D) {}

所以，在这个函数的开头，我必须做参数检查：

• 是否缺少一些必需的参数
• 是否所有参数都遵循规则：class(data type) 或 A 必须在 [1:3] 范围内

但是关于这些参数，有些是必须的，有些不是，而且每个参数的class(data type)必须是我想要的，例如：numeric ，逻辑，等等...

为了做到这一点，我做了这样的事情：

if(A Follow_the_rule){}
if(B Follow_the_rule){}
if(C Follow_the_rule){}
...

关于上面的代码，需要很多 if 语句，我认为这不是检查参数的最佳方式。

那么有没有更好的方法来检查 R 中的参数？

任何帮助将不胜感激。

Answer 1

看看?stopifnot，这正是你想要的。它检查其中的条件，如果没有给出条件，则停止。与if 相同，您可以将条件与&& 或|| 和& 和| 连接起来。查看信息，例如?"&"。 all 或 any 也可能有助于检查给定向量的所有元素是否分别满足条件或任何条件。一些例子：

foo <- function(A, B, C){
  stopifnot(!missing(C), !missing(B), !missing(A)) ##A, B, C not missing, then continue
  stopifnot(class(B)=="matrix") ## B is a matrix, then continue
  stopifnot(class(B)==class(C), all(B > C)) ## class B is class C and all elements of B are greater than C
  stopifnot((length(A)>1 && !any(is.na(A))) || all(A==0)) ## (A has more than 1 element and no element is NA) or all elements of A are 0.
  stopifnot(all(A > 2), all(A < 10)) ## all elements of A are between 2 and 10, else stop.

  #... further code
}

上述条件可能不适合这种组合，但我认为有足够的例子让你适应你的问题。当然你可以写一个stopifnot，但是如果有很多条件，将条件分组更有用，因为停止函数的条件被打印为错误代码。所以stopifnot越多，得到的错误信息就越准确。