R:重写" ["对于矩阵?

时间:2017-07-29 10:04:17

标签: r class subset r-s3

我正在尝试附加"矩阵"类反过来覆盖" ["的默认行为。代码示例如下:

annMatrix <- function(mat=NULL, rowAnn=NULL, colAnn=NULL) {
  if(is.null(mat)) mat <- matrix(nrow=0, ncol=0)
  mat <- as.matrix(mat)
  if(is.null(rowAnn)) rowAnn <- data.frame(row.names=seq_len(nrow(mat)))
  if(is.null(colAnn)) colAnn <- data.frame(row.names=seq_len(ncol(mat)))
  rowAnn <- data.frame(rowAnn, stringsAsFactors=FALSE)
  colAnn <- data.frame(colAnn, stringsAsFactors=FALSE)
  stopifnot(nrow(mat)==nrow(rowAnn) & ncol(mat)==nrow(colAnn))
  attr(mat, "colAnn") <- colAnn
  attr(mat, "rowAnn") <- rowAnn
  class(mat) <- append(class(mat), "annMatrix")
  mat
}

`[.annMatrix` <- function(annMat, rowExpr=NULL, colExpr=NULL) {
  stopifnot(is.valid.annMatrix(annMat))
  rowExpr <- eval(substitute(list(rowExpr)), attr(annMat, "rowAnn"), parent.frame())
  colExpr <- eval(substitute(list(colExpr)), attr(annMat, "colAnn"), parent.frame())
  indsR   <- unlist(rowExpr)
  indsC   <- unlist(colExpr)
  if(is.null(indsR)) indsR <- seq_len(nrow(annMat))
  if(is.null(indsC)) indsC <- seq_len(ncol(annMat))
  attr(annMat, "rowAnn") <- attr(annMat, "rowAnn")[indsR,,drop=FALSE]
  attr(annMat, "colAnn") <- attr(annMat, "colAnn")[indsC,,drop=FALSE]
  annMat <- unclass(annMat)
  annMat <- annMat[indsR,indsC,drop=FALSE]
  class(annMat) <- append(class(annMat), "annMatrix")
  annMat
}

基本思想是在子集化后使矩阵保留其特定属性。

但是我遇到了一个问题:

如何写&#34; [&#34;以这样的方式运行,当使用和不使用逗号调用时,它的行为会有所不同:

annMat[i]
annMat[i,]

作为默认&#34; [&#34;对于矩阵似乎也是如此。

我打算默认将第二个参数设置为某个值,但由于添加了逗号,该值不会更改。

0 个答案:

没有答案