0

我正在尝试创建自己的功能，我想同时返回POSIXct和numeric值。我发现唯一的解决方法是使用unnest()afterfeatures函数。我们可以简化这个工作流程吗？我想在我的包中包含功能，我不想让用户unnest每次都在features.

示例代码

library(feasts, quietly = T, warn.conflicts = F)
library(dplyr, quietly = T, warn.conflicts = F)
library(tidyr, warn.conflicts = F, quietly = T)

  df <- data.frame(datetime = seq(c(ISOdate(2000,3,20)),
                                  by = "hour",
                                  length.out = 30),
                   group = c(rep("A", 15), rep("B", 15)),
                   x = rnorm(30))
  
  feat_event <- function(x){
    
    start <- dplyr::first(x)
    end <- dplyr::last(x)
    length <- difftime(end, start, units = "h")
    
    output <- list(
      start = unname(start),
      end = unname(end),
      length = unname(length)
    )
    
    output
  }
  
  feat_event(df$datetime)
#> $start
#> [1] "2000-03-20 12:00:00 GMT"
#> 
#> $end
#> [1] "2000-03-21 17:00:00 GMT"
#> 
#> $length
#> Time difference of 29 hours
  
  df %>% 
    as_tsibble(key = group, index = datetime) %>% 
    features(datetime, feat_event) %>% 
    unnest()
#> Warning: `cols` is now required when using unnest().
#> Please use `cols = c(start, end, length)`
#> # A tibble: 2 x 4
#>   group start               end                 length  
#>   <chr> <dttm>              <dttm>              <drtn>  
#> 1 A     2000-03-20 12:00:00 2000-03-21 02:00:00 14 hours
#> 2 B     2000-03-21 03:00:00 2000-03-21 17:00:00 14 hours

由reprex 包于 2021-03-17 创建(v1.0.0)

4

1 回答 1

1

返回一个小标题而不是列表。

feat_event <- function(x){
  
  start <- dplyr::first(x)
  end <- dplyr::last(x)
  length <- difftime(end, start, units = "h")
  
  output <- tibble(
    start = unname(start),
    end = unname(end),
    length = unname(length)
  )
  
  output
}

feat_event(df$datetime)
# A tibble: 1 x 3
#  start               end                 length  
#  <dttm>              <dttm>              <drtn>  
#1 2000-03-20 12:00:00 2000-03-21 17:00:00 29 hours

df %>% 
   as_tsibble(key = group, index = datetime) %>% 
   features(datetime, feat_event)

# A tibble: 2 x 4
#  group start               end                 length  
#  <chr> <dttm>              <dttm>              <drtn>  
#1 A     2000-03-20 12:00:00 2000-03-21 02:00:00 14 hours
#2 B     2000-03-21 03:00:00 2000-03-21 17:00:00 14 hours
于 2021-03-17T10:42:10.010 回答