假设我有这样的功能:
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了。
答案 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