R S4 设置方法 '['区分缺少的参数？

Question

对

标题表示歉意，不知道如何表达这个问题。

如果我想在类上创建setMethod，如何区分mat[i,]和mat[i]的相似情况？

我知道对于前者，我可以使用：

setMethod("[",
    signature(x = "foo", j = "missing", drop = "missing"),
    function(x,i,j,drop) return(myFunc(x,i))
)

如何设置一种方法来区分后者，我不想像base matrix类那样选择行而是选择元素？

查看?[文档，我希望以下内容可以工作：

setMethod("[",
    signature(x = "foo", i = "numeric"),
    function(x,i,j,drop) return(myFunc(x,i))
)

但它与缺少j和drop的任何先前定义的方法冲突。

想法与

mat <- matrix(seq(9), 3, 3)
mat[c(1,3),]
1 4 7
3 6 9
mat[c(1,3)]
[1] 1 3

Answer 1

你会在矩阵包中找到很多例子。它使用 S4 并为矩阵实现新的类和方法。据我所知，没有办法在签名中声明您要查找的内容。相反，您必须使用函数nargs来区分mat[1]和mat[1, ]。下面是如何执行此操作的示例：

setClass("foo", slot = c(mat = "matrix"))
setMethod(
  "[",
  signature(x = "foo", i = "missing", j = "missing", drop = "missing"),
  function(x, i, j, drop = FALSE) {
    x
  }
)
setMethod(
  "[",
  signature(x = "foo", i = "numeric", j = "missing", drop = "missing"),
  function(x, i, j, ..., drop) {
    if (nargs() == 3) x@mat[i, ]
    else x@mat[i]
  }
)
setMethod(
  "[",
  signature(x = "foo", i = "numeric", j = "numeric", drop = "missing"),
  function(x, i, j, ..., drop) {
    x@mat[i, j]
  }
)
mat <- new("foo", mat = matrix(seq(9), 3, 3))
mat[]
mat[2:5]
mat[1:2, ]
mat[1:2, 2]

但是，如果您直接扩展基类"matrix"（或矩阵包中的"矩阵"）并执行类似操作会更容易

setClass("Matrix", contains = "matrix")

因为您可以免费获得这些方法。例如，请注意，在上面的实现中，您仍然需要注意参数drop。基本上你必须重新实现已经存在的东西。