如何從R中的父函式訪問變數

我對使用父函式中的不可變變數的優雅方式感興趣。

讓我們看一個例子。

我有帶有周數及其相應值的資料框。

df <- tribble(
  ~week, ~count,
  1, 99.6,
  2, 116, 
  3, 107, 
  4, 125, 
  5, 126, 
  6, 131, 
  7, 149, 
  8, 130, 
  9, 111,
  48, 43.4,
  49, 136, 
  50, 133,
  51, 115, 
  52, 93.3
  )

以下函式將每周的第一個和最后一個日期添加到資料框列中。

  add_week_limit_dates <- function(df){
    year_change <- ifelse(any(df$week < 25) && any(df$week > 38), TRUE, FALSE)
    df %>% 
      mutate(week_start = week_to_date(week, 1, year_change),
             week_end = week_to_date(week, 7, year_change))
  }
  
  week_to_date <- function(week, day_of_week, year_change){
    paste0(get_year(week, year_change), 
           "-W", ifelse(week < 10, paste0("0", as.character(week)), week), 
           "-", day_of_week) %>% ISOweek2date()
  }
  
  get_year <- function(week, year_change){
    if(isTRUE(year_change)){
      ifelse(week < 38, year(Sys.Date()), year(Sys.Date()) - 1)
    } else{
      year(Sys.Date())
    }
  }

> add_week_limit_dates(df)
# A tibble: 14 × 4
    week count week_start week_end  
   <dbl> <dbl> <date>     <date>    
 1     1  99.6 2022-01-03 2022-01-09
 2     2 116.  2022-01-10 2022-01-16
 3     3 107.  2022-01-17 2022-01-23
 4     4 125.  2022-01-24 2022-01-30
 5     5 126.  2022-01-31 2022-02-06
 6     6 131.  2022-02-07 2022-02-13
 7     7 149.  2022-02-14 2022-02-20
 8     8 130.  2022-02-21 2022-02-27
 9     9 111.  2022-02-28 2022-03-06
10    48 43.4  2021-11-29 2021-12-05
11    49 136.  2021-12-06 2021-12-12
12    50 133.  2021-12-13 2021-12-19
13    51 115.  2021-12-20 2021-12-26
14    52 93.3  2021-12-27 2022-01-02

所以，我的問題是：是否可以不明確設定公共變數year_change并強制子函式在父函式中搜索它？我夢想著下一條路

df %>% 
  mutate(week_start = week_to_date(week, 1),
         week_end = week_to_date(week, 7))

UPD - 解決方案

按照建議，我訪問了parent.frame()，但請注意，當您在 dplyr 的方法中使用它時，它不是預期的

 add_week_limit_dates <- function(df){
    year_change <- ifelse(any(df$week < 25) && any(df$week > 38), TRUE, FALSE)
    df$week_start <- week_to_date(df$week, 1)
    df$week_end <- week_to_date(df$week, 7)
    df
  }
  
  week_to_date <- function(week, day_of_week){
    paste0(get_year(week, parent.frame()$year_change), 
           "-W", ifelse(week < 10, paste0("0", as.character(week)), week), 
           "-", day_of_week) %>% ISOweek2date()
  }
  
...

uj5u.com熱心網友回復：

您可以通過被呼叫函式的父框架訪問呼叫函式的范圍：

mother <- function(){
  year_change = 8
  child()
}

child <- function(){
  print(parent.frame()$year_change)
}

...或在函式的直接父環境之一中定義它們：

mother <- function(){ ## now the immediate parent environment
  year_change = 8
  child <- function(){
    print(year_change)
  }
  child()
}

...或在全球環境中：

year_change = 8 ## the global environment

mother <- function(){
  print(paste("mom's idea of year_change is:", year_change))
  child <- function(){
    print(paste("kid's idea of year_change is:", year_change))
  }
  child()
}

請注意，如果變數具有相同的名稱，則直接父環境與更遠的祖先環境相匹配。

范圍的詳細資訊，例如 H. Wickham: Advanced R

編輯

可恥地忘記了利用省略號...引數傳遞變數的可能性。

例子

mother_says <- function(...){
    tell_child(...)
}

tell_child <- function(...){
    args <- list(...)
    print(paste("Mom says:", args$message_to_daughter))
    tell_grandchild(...)
}

tell_grandchild <- function(...){
    args <- list(...)
    print(paste("Granny says:", args$message_to_grandchild))
}

示例的輸出：

## > mother_says(message_to_daughter = "Hi, daughter, how're you?",
##          message_to_grandchild = "Grandchild, sweetie!"
##          )
##
## [1] "Mom says: Hi, daughter, how're you?"
## [1] "Granny says: Grandchild, sweetie!"

標籤：r

上一篇：我如何在R中按ID分組并用na.rm=TRUE進行平均匯總

下一篇：在dbplyr管道中使用類似grepl的字串匹配