Início / coding / Perguntas / 79598831
Accepted
droubi
Asked: 2025-04-30 00:36:56 +0800 CST

Programação funcional com data.table

  • 772

Quero criar uma função simples em R que funcione com data.frames/tibbles ou com data.table. Então, criei um novo método:

pesquisarComposicao <- function(BASE, ...) {
  UseMethod("pesquisarComposicao")
}

Então criei o método para data.frames, que demonstra uma função de busca bem simples. Funciona muito bem.

pesquisarComposicao.data.frame <- function(BASE, TERMO, CAMPO = DESCRICAO) {`

  BASE |>
    filter(grepl(pattern = {{TERMO}}, x = {{CAMPO}}))
}

Mas o método data.table abaixo não está funcionando (eu estava seguindo as instruções aqui ):

pesquisarComposicao.data.table <- function(BASE, TERMO,
                                           CAMPO = "DESCRICAO DA COMPOSICAO") {

  # filter_col <- NULL
  # filter_val <- NULL

  BASE[filter_col %like% filter_val,
     env = list(
       filter_col = CAMPO,
       filter_val = I(TERMO)
     ), verbose = FALSE]
}

Tentei inserir filter_col = NULLe filter_val = NULLpara evitar o seguinte erro:

Error in pesquisarComposicao.data.table(BASE = sinteticoSINAPI, TERMO = "PEITORIL") : object 'filter_col' not found

Mas então obtive outro erro:

Error in grepl(pattern, vector, ignore.case = ignore.case, fixed = fixed, : invalid 'pattern' argument

Acho estranho, porque quando eu ainda não usava métodos, mas programei uma função que deveria funcionar para data.tables e data.frames simples (verificando internamente se o objeto era de um tipo ou outro), este trecho de código acima funcionava bem (e eu não precisava do filter_col = NULLand filter_val = NULL). Por que funcionou para uma única função, mas não funcionou como um método?

1 respostas

  1. Best Answer

    Eu penso

    library(data.table)
pesquisarComposicao <- function(BASE, ...) {
  UseMethod('pesquisarComposicao')
}
pesquisarComposicao.data.table  <- function(BASE, TERMO, CAMPO) {
  BASE[grepl(TERMO, BASE[[CAMPO]])] # non-lazy 
  
  # BASE[get(CAMPO) %like% TERMO] # get
  # BASE[grepl(TERMO, get(CAMPO))] # get
  
  # as of data.table 1.15.0 there is substitute2: 
  # BASE[grepl(pattern, x), env = I(list(pattern=TERMO, x=as.name(CAMPO)))]
  # BASE[grepl(pattern, x), env = list(pattern=I(TERMO), x=CAMPO)]
  # where `grepl`/%like% (which is like()) could have been passed as well. 
  # the docs are hard to read. 
}

    faz o que você procura.

    > pesquisarComposicao(BASE=dt, TERMO='DESCRICAO DA COMPOSICAO', CAMPO='DESCRICAO')
       ID                        DESCRICAO
   <char>                           <char>
1:      A DESCRICAO DA COMPOSICAO DE PRODU
2:      C DESCRICAO DA COMPOSICAO COMPLETA

    Dados de amostra

    já que o OP não fornece alguns.

    dt = data.table::data.table(ID = LETTERS[1:4], DESCRICAO = c('DESCRICAO DA COMPOSICAO DE PRODU', 'INGREDIENTES NATURAIS', 'DESCRICAO DA COMPOSICAO COMPLETA', 'OUTRA COISA'))
    • 2

relate perguntas