制作列表的子类

时间:2014-06-18 11:09:54

标签: r r-s3

我有以下代码

obj <- list(list(a=4,f=5,g=5),list(a=44,f=54,g=54))
class(obj) <- "mysubclass"

class(obj[1])
class(obj[2])
class(obj[1:2])
class(obj)

导致:

> class(obj[1])
[1] "list"
> class(obj[2])
[1] "list"
> class(obj[1:2])
[1] "list"
> class(obj)
[1] "mysubclass"

通过子集化不会失去课程的正确解决方案是什么?例如,class(obj[1:2])会产生mysubclass并仍然表现为列表。

1 个答案:

答案 0 :(得分:5)

问题在于通用[方法正在剥离class属性。为避免这种情况,您可以为mysubclass定义自己的通用,即

## Answer suggested by Petr Matousu
## Based on simple.list method
'[.mysubclass' = function(x, i, ...) {
    structure(NextMethod("["), class = class(x))
 }

或者

'[.mysubclass' = function(x, i, j, ..., drop=TRUE) {
    ## Store the class attribute
    cl = class(x)
    ## Pass your object to the next method
    y = NextMethod('[')
    ## Update the class and return
    class(y) = cl
    y
}

您的示例现在可以按预期工作了。您还应该看看:

  • help('[')
  • methods('[')