Question

我已经定义了使用test类打印矢量的方法：

print.test <- function(x,  ...) {
    x <- formatC(
        as.numeric(x),
        format = "f",
        big.mark = ".",
        decimal.mark = ",",
        digits = 1
        )
    x[x == "NA"] <- "-"
    x[x == "NaN"] <- "-"
    print.default(x)
}

适用于以下情况

a <- c(1000.11, 2000.22, 3000.33)
class(a) <- c("test", class(a))
print(a)
[1] "1.000,11" "2.000,22" "3.000,33"

这也有效：

round(a)
[1] "1.000,0" "2.000,0" "3.000,0"

这不是：

median(a)
[1] 2000.22
class(median(a))
[1] "numeric"

现在我的问题是：我需要为此类使用中位数来编写自定义方法吗？如果是这样的话，它是什么样？还是有另一种方式（因为我只是想让此类以某种格式打印数据）？

Answer 1

问题是median.default返回了类numeric的对象，因此自动打印返回的对象不会调用您的自定义print方法。
下面将这样做。

median.test <- function(x, na.rm = FALSE, ...){
    y <- NextMethod(x, na.rm = na.rm, ...)
    class(y) <- c("test", class(y))
    y
}

median(a)
#[1] "2.000,2"

对于NA值的处理，我将首先为基本R函数定义另一种方法。如果经常使用类test的对象，则不是必须的，但是可以节省一些代码行。

c.test <- function(x, ...){
    y <- NextMethod(x, ...)
    class(y) <- c("test", class(y))
    y
}


b <- c(a, NA)
class(b)
#[1] "test"    "numeric"

median(b)
#[1] "-"

median(b, na.rm = TRUE)
#[1] "2.000,2"

编辑。

以下内容按照OP在注释中的要求，定义了泛型函数wMedian，默认方法和类"currency"对象的方法。

请注意，必须有一个方法print.currency，由于它与上面的print.test完全相同，因此我没有重新定义。至于其他方法，我借助新功能as.currency使它们更简单。

median.currency <- function(x, na.rm = FALSE, ...){
  y <- NextMethod(x, na.rm = na.rm, ...)
  as.currency(y)
}

c.currency <- function(x, ...){
  y <- NextMethod(x, ...)
  as.currency(y)
}

as.currency <- function(x){
  class(x) <- c("currency", class(x))
  x
}

wMedian <- function(x, ...) UseMethod("wMedian")
wMedian.default <- function(x, ...){
    matrixStats::weightedMedian(x, ...)
}

wMedian.currency <- function(x, w = NULL, idxs = NULL, na.rm = FALSE, interpolate = is.null(ties), ties = NULL, ...) {
  y <- NextMethod(x, w = w, idxs = idxs, na.rm = na.rm, interpolate = interpolate, ties = ties, ... ) 
  as.currency(y)
}


set.seed(1)
x <- rnorm(10)
wMedian(x, w = (1:10)/10)
#[1] 0.4084684
wMedian(as.currency(x), w = (1:10)/10)
#[1] "0,4"

r如何保留自定义类的打印方法

1 个答案: