Início / coding / Perguntas / 79076724
Accepted
user18894435
Asked: 2024-10-11 11:27:53 +0800 CST

Execute uma lista de fórmulas lambda estilo purrr em um quadro de dados

  • 772

Os dados do brinquedo a seguir têm 5 variáveis, X1para X5.

set.seed(123)
df <- data.frame(matrix(rnorm(500), 100, 5))

Quero executar operações específicas em variáveis ​​específicas, usando uma lista nomeada de fórmulas lambda no estilo purrr

fun_list <- list(
  X2 = ~ quantile(.x, c(0.1, 0.9), na.rm = TRUE),
  X4 = ~ fivenum(.x, na.rm = TRUE)
)

Como posso aplicar fun_listmeu dfde acordo com seus nomes de variáveis?

Eu sei rlang::as_function()que pode converter uma fórmula estilo purrr em uma função R. Mas acho que há alguma função que é capaz de lidar com fórmulas estilo purrr intrinsecamente. Seu uso pode ser

execute(fun_list, environment = df)

A saída esperada é

$X2
      10%       90% 
-1.289408  1.058432 

$X4
[1] -2.465898194 -0.737146704 -0.003508661  0.693634712  2.571458146

2 respostas

  1. Best Answer

    Uma solução alternativa é usar um map, que pode receber uma fórmula no estilo purrr como entrada e evitar o uso de rlang::as_function().

    library(purrr)

imap(fun_list, \(f, var) map(df[var], f)[[1]])

# $X2
#       10%       90% 
# -1.289408  1.058432 
# 
# $X4
# [1] -2.465898194 -0.737146704 -0.003508661  0.693634712  2.571458146

    ou resumidamente, imap(fun_list, ~ map(df[.y], .x)[[1]]).

    • 2

  2. 1) Aqui está uma solução base R. Primeiro, criamos uma função, fo2fun, que aceita uma fórmula e gera a função correspondente. Então, executeé uma função com um corpo de uma instrução usando Mappara aplicá-la a cada fórmula e nome/índice de lista.

    fo2fun <- function(formula) {
    f <- function(.x) {}
    body(f) <- formula[[2]]
    environment(f)  <- environment(formula)
    f
}

execute <- function(funs, envir = parent.frame()) {
  Map(\(fo, index) fo2fun(fo)(envir[[index]]), funs, names(fun_list))
}

# test
expected <- list(
  X2 = quantile(df$X2, c(0.1, 0.9), na.rm = TRUE), 
  X4 = fivenum(df$X4, na.rm = TRUE)
)

execute(fun_list, df) |> identical(expected)
## [1] TRUE

execute(fun_list, list2env(df)) |> identical(expected)
## [1] TRUE

list2env(df, .GlobalEnv)
execute(fun_list) |> identical(expected)
## [1] TRUE

    2) É o mesmo que (1), exceto que usamos match.funfndo pacote gsubfn no lugar de fo2fun.

    Com essa abordagem, o argumento formal não é restrito a ser .x, mas match.funfnassume que qualquer variável livre encontrada na fórmula é o argumento. Opcionalmente, especifique a variável do argumento no lado esquerdo da fórmula. Esta última sintaxe deve ser usada se houver variáveis ​​livres sem argumento na fórmula para distinguir o argumento, mas também pode ser usada mesmo se não houver.

    library(gsubfn)

fun_list2 <- list(
  X2 = ~ quantile(var, c(0.1, 0.9), na.rm = TRUE),
  X4 = x ~ fivenum(x, na.rm = TRUE)
)

execute <- function(funs, envir = parent.frame()) {
  Map(\(fo, index) match.funfn(fo)(envir[[index]]), funs, names(fun_list))
}

# test

execute(fun_list, df) |> identical(expected)
## [1] TRUE

execute(fun_list2, df) |> identical(expected)
## [1] TRUE

    Observação

    Entrada da pergunta:

    set.seed(123)
df <- data.frame(matrix(rnorm(500), 100, 5))

fun_list <- list(
  X2 = ~ quantile(.x, c(0.1, 0.9), na.rm = TRUE),
  X4 = ~ fivenum(.x, na.rm = TRUE)
)
    • 2

relate perguntas