oop - R中的矩阵丢失类属性

Question

考虑以下代码：

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"

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

score 7 · Accepted Answer

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
}

1 回答 1