如何定义R函数的参数类型？

Question

我正在写一个R函数，我想确保R函数的参数是某个类（例如“矩阵”）的参数。

这样做的最好方法是什么？

说我有一个“foo”函数，它计算矩阵的逆：

foo <- function(x) 
{ 
    # I want to make sure x is of type "matrix" 
    solve(x) 
} 

我怎么能说 - 你可能用C - function(matrix x)来表示“x必须matrix类型，如果不是，那么返回一个错误“？

Answer 1

您可以检查它与is.matrix或者与as.matrix参数通过后，将其转换矩阵：

foo <- function(x) 
{ 
    if(!is.matrix(x)) stop("x must be a matrix") 
    # I want to make sure x is of type "matrix" 
    solve(x) 
} 

Answer 2

只是为了完整性：除了

is.matrix(foo) 

你也可以测试为

class(foo) == "matrix" 

这也适用于非标准的没有is.foo()函数。

Answer 3

stopifnot(is.matrix(x))