Início / coding / Perguntas / 78310377
Accepted
thothal
Asked: 2024-04-11 20:35:46 +0800 CST

Nome da saída do pmap no tibble

  • 772

Quero fazer um loop nas linhas de a tibble, uma coluna deve conter o nome que o slot correspondente deve ter na saída e as outras colunas são os argumentos da função.

Eu sei como resolver isso, mas queria saber se existe alguma possibilidade de fazer a nomenclatura dentro da pmapfunção?

library(purrr)
library(tibble)

params <- tibble(nm = LETTERS[1:2], x = 1:2, y = 2:3)

## this has one addional level of nesting
pmap(params, function(nm, x, y) {
  set_names(list(x + y), nm)
}) # %>% flatten would remove the unnecessary level

## correct format but mixes tidyverse syntax with base syntax
map(split(params, params$nm), ~ .x$x + .x$y)

## correct format & tidyverse syntax but naming must happen **outside**
pmap(params, function(nm, x, y) {
  x + y
}) %>%
  set_names(params$nm)

Resultado esperado como na versão 2 ou 3, com tidyverseestilo, mas sem a necessidade de renomear o resultado posteriormente.

3 respostas

  1. E se algo assim estiver totalmente em tidyversesintaxe?

    params %>% 
  transmute(lst = as.list(x+y) %>% `names<-`(nm)) %>% 
  pull(lst)
    • 3
  2. Best Answer

    A maneira como pmapfunciona é que ele pega os nomes dos nomes do primeiro argumento, então isso não faz o pós-processamento da pmapsaída, mas adiciona esses nomes antes da execução pmappara que a entrada corresponda ao que ele espera.

    params %>%
  mutate(nm = set_names(nm)) %>% 
  pmap(function(nm, x, y) {
    x + y
})

    dando

    $A
[1] 3

$B
[1] 5
    • 2

  3. Outra opção com pmap:

      pmap(params, function(nm, x, y) {
    as.list(x + y) %>% `names<-`(nm) 
  }) %>% 
    unlist(recursive = F)
    • 1

relate perguntas