Question

我创建了一个名为time的类。这是一个虚拟示例，返回秒为分钟。效果很好，但是print.time函数不在tbl内部使用。

任何想法如何在tbl内调整显示？

致谢

在此处查看reprex

library(dplyr)
#> 
#> Attachement du package : 'dplyr'
#> The following objects are masked from 'package:stats':
#> 
#>     filter, lag
#> The following objects are masked from 'package:base':
#> 
#>     intersect, setdiff, setequal, union
as.time <- function(x){
  class(x)<-"time"
  x
}
print.time<-function(x,...){
  print.default(unclass(x/60)) 
  invisible(x)
}


60 %>% as.time() %>% print()
#> [1] 1


tribble(~a,~time,
        "a",123,
        "b",234,
        "c",456
) %>% mutate(time = as.time(time))
#> # A tibble: 3 x 2
#>   a      time
#>   <chr> <dbl>
#> 1 a       123
#> 2 b       234
#> 3 c       456

^{由reprex package（v0.2.1）于2019-02-07创建}

Answer 1

快速又肮脏的黑客攻击

您可以重写print.tbl方法以执行自定义操作，然后调用tibble:::print.tbl函数：

library(tidyverse)

as.time <- function(x){
  class(x)<-"time"
  x
}

is.time <- function(x) "time" %in% class(x)

res <- tribble(~a,~time,
               "a",123,
               "b",234,
               "c",456
) %>% mutate(time = as.time(time))

print.tbl <- function(x, ...){
  res <- mutate_if(x, is.time, ~ .x / 60)
  tibble:::print.tbl(res, ...)
}
res
#> # A tibble: 3 x 2
#>   a      time
#>   <chr> <dbl>
#> 1 a      2.05
#> 2 b      3.9 
#> 3 c      7.6

# Or if you want to make it ok for CRAN 

print.tbl <- function(x, ...){
  res <- mutate_if(x, is.time, ~ .x / 60)
  print_tbl <- getFromNamespace("print.tbl", "tibble")
  print_tbl(res, ...)
}
res
#> # A tibble: 3 x 2
#>   a      time
#>   <chr> <dbl>
#> 1 a      2.05
#> 2 b      3.9 
#> 3 c      7.6

^{由reprex package（v0.2.1）于2019-02-07创建}

编辑：正确的方法

您需要定义几种方法：print和format，pillar_shaft和type_sum。

library(tibble)
library(pillar)
time <- function(x) {
  as_time(x)
}

as_time <- function(x) {
  structure(x, class = "time")
}

c.time <- function(x, ...) {
  as_time(NextMethod())
}

`[.time` <- function(x, i) {
  as_time(NextMethod())
}


format.time <- function(x, ...) {
  ret <- unclass(x / 60)
  format(ret)
}

print.time <- function(x, ...) {
  cat(format(x), sep = "\n")
  invisible(x)
}

time(360)
#> 6

type_sum.time <- function(x) {
  "time"
}

pillar_shaft.time <- function(x, ...) {
  out <- format(x)
  pillar::new_pillar_shaft_simple(out)
}


data <- tibble(
  loc   = time(360)
)

data
#> # A tibble: 1 x 1
#>   loc   
#>   <time>
#> 1 6

^{由reprex package（v0.2.1）于2019-02-07创建}

有关更多信息，请参见：https://cran.r-project.org/web/packages/tibble/vignettes/extending.html

在print.tbl中使用个人打印方法

1 个答案:

快速又肮脏的黑客攻击

编辑：正确的方法