R中的对象可以有多个类吗?

时间:2013-10-12 15:28:38

标签: r oop

假设我有这样的功能:

myf = function(x) {
res = dostuff(x)
res # this is a data.frame
}

我想对res做一些特别的事,例如, 我想制作一些通用函数,如print.myf, summary.myf, ... 所以我可以继续给它上课:

myf = function(x) {
res = dostuff(x)
class(res) = 'myf'
res
}

但是这样我就不能再将它用作data.frame了。

2 个答案:

答案 0 :(得分:2)

是的,我的标准(简单)示例是

R> now <- Sys.time()
R> class(now)
[1] "POSIXct" "POSIXt" 
R> class(as.POSIXlt(now))
[1] "POSIXlt" "POSIXt" 
R> 

这也是使用inherits("someClass")进行测试的专业提示背后的原因,而不是测试class(obj)=="someClass"的结果。

答案 1 :(得分:2)

以下是rle的示例。 rle函数会创建list,但它没有类list,因此as.data.frame等方法无法“开箱即用”。因此,我更改了最后一行,将"list"添加为其中一个类。

x <- c(1, 1, 1, 1, 2, 2, 2, 3, 3, 3, 3, 3, 3, 3)

rle2 <- function (x)
{
  if (!is.vector(x) && !is.list(x))
    stop("'x' must be an atomic vector")
  n <- length(x)
  if (n == 0L)
    return(structure(list(lengths = integer(), values = x),
                     class = "rle"))
  y <- x[-1L] != x[-n]
  i <- c(which(y | is.na(y)), n)

  ## THE FOLLOWING IS FROM THE BASE RLE. NOTICE ONLY ONE CLASS...
  # structure(list(lengths = diff(c(0L, i)), values = x[i]), 
  #  class = "rle")

  structure(list(lengths = diff(c(0L, i)), values = x[i]),
            class = c("rle", "list"))
}

如您所见,我只是更改了class。功能的其余部分是相同的。

rle(x)
# Run Length Encoding
#   lengths: int [1:3] 4 3 7
#   values : num [1:3] 1 2 3
data.frame(rle(x))
# Error in as.data.frame.default(x[[i]], optional = TRUE, stringsAsFactors = stringsAsFactors) : 
#   cannot coerce class ""rle"" to a data.frame
rle2(x)
# Run Length Encoding
#   lengths: int [1:3] 4 3 7
#   values : num [1:3] 1 2 3
data.frame(rle2(x))
#   lengths values
# 1       4      1
# 2       3      2
# 3       7      3

当然,如果我们知道这一点,我们也可以明确指定我们的方法,如果我们知道一个存在:

as.data.frame.list(rle(x))
#   lengths values
# 1       4      1
# 2       3      2
# 3       7      3