矩阵丢失R中的类属性

时间:2011-09-23 17:38:47

标签: oop r r-s3

请考虑以下代码:

> A <- matrix(1:12, ncol=4)
> colnames(A) <- letters[1:4]
> class(A) <- c("foo", "matrix")

当A是子集时,它会丢失“foo”类标签:

> class(A[1:2,])
[1] "matrix"

向量也是如此。然而,data.frames不会发生同样的情况:

> B <- as.data.frame(A)
class(B) <- c("foo", "data.frame")
> class(B[1:2,])
[1] "foo"        "data.frame"

通常,将泛型函数应用于对象会保留class属性。不适用于矩阵/数字/整数对象。为什么?并且可以避免这种行为吗?

1 个答案:

答案 0 :(得分:7)

data.frames有自己的子集方法[.data.frame,它为您处理类。我不确定为什么Primitive不保留类,但是创建自己的子集方法非常简单。

`[.foo` <- function(x, i, j, ...) {
  y <- unclass(x)[i,j,...]
  class(y) <- c("foo",class(y))
  y
}
class(A[1:2,])
# [1] "foo"    "matrix"

编辑:
正如其他人所提到的,这里应该使用NextMethod

`[.foo` <- `[.bar` <- function(x, i, j, ...) {
  y <- NextMethod(.Generic)
  class(y) <- .Class
  y
}