R - Использование purrr :: pmap () для построчной итерации - PullRequest
0 голосов
/ 02 марта 2019

Я пытаюсь понять, как работает pmap.В таблице ниже приведен список-столбец values.Я хотел бы создать новый столбец New, который зависит от того, являются ли соответствующие элементы в столбце values NULL.Поскольку is.null не векторизован, я сначала подумал использовать rowwise(), прежде чем встретить pmap().

Использование rowwise() до mutate() дает мне желаемый результат, как показано ниже:

tbl = as.data.frame(do.call(rbind, pars)) %>%
  rowwise() %>%
  mutate(New = ifelse(is.null(values), paste(id, default), paste(id, values, collapse=", ")))

> tbl
Source: local data frame [2 x 6]
Groups: <by row>

# A tibble: 2 x 6
  id        lower     upper     values     default   New        
  <list>    <list>    <list>    <list>     <list>    <chr>        
1 <chr [1]> <dbl [1]> <dbl [1]> <NULL>     <dbl [1]> a 5          
2 <chr [1]> <NULL>    <NULL>    <list [3]> <chr [1]> b 0, b 1, b 2

Однако, pmap() не:

tbl = as.data.frame(do.call(rbind, pars)) %>%
      mutate(New = pmap(., ~ifelse(is.null(values), paste(id, default), paste(id, values, collapse=", "))))

> tbl
  id lower upper  values default                         New
1  a     1    10    NULL       5 a NULL, b list("0", "1", "2")
2  b  NULL  NULL 0, 1, 2       1 a NULL, b list("0", "1", "2")

Это кажется работать, если я использую анонимную функцию вместо тильды:

tbl = as.data.frame(do.call(rbind, pars)) %>%
  mutate(Value = pmap(., function(values, default, id, ...) ifelse(is.null(values), paste(id, default), paste(id, values, collapse=", "))))

> tbl
  id lower upper  values default         Value
1  a     1    10    NULL       5           a 5
2  b  NULL  NULL 0, 1, 2       1 b 0, b 1, b 2

Но я не понимаю, почему не работает версия тильды?Я бы предпочел не указывать аргументы полностью, так как мне нужно отобразить функцию на несколько столбцов.Куда я иду не так?

1 Ответ

0 голосов
/ 05 сентября 2019

Я собирался задать очень похожий вопрос к этому.По сути, спрашивая, как использовать pmap в mutate, не используя имена переменных более одного раза.Вместо этого я опубликую его как «ответ» здесь, поскольку он включает в себя предисловие и ряд опций, которые я нашел, ни один из которых не является полностью удовлетворительным для меня.Надеюсь, кто-то еще сможет ответить, как это сделать.

Я часто хочу использовать purrr::pmap внутри dplyr::mutate при работе с data.frame со списком-столбцами.Иногда это включает многократное повторение имен переменных.Я хотел бы иметь возможность сделать это более кратко, используя анонимную функцию, чтобы переменные использовались только один раз, когда передаются в pmap '.f аргумент.

Примите этот небольшой набор данных какпример:

library('dplyr')
library('purrr')

df <- tribble(
  ~x,   ~y,      ~z,         
  c(1), c(1,10), c(1, 10, 100),
  c(2), c(2,20), c(2, 20, 200),
)

Скажем, функция, которую я хочу применить к каждой строке:

func <- function(x, y, z){c(sum(x), sum(y), sum(z))}

На практике функция будет более сложной, с большим количеством переменных.Функция нужна только один раз, поэтому я бы предпочел не называть ее явно и не засорить мой сценарий и мою рабочую среду.

Вот варианты.Каждый создает точно такой же data.frame, но по-своему.Причина включения avg`` will be come clear. Note I'm not considering position matching using .. 1 , .. 2` и т. Д., Поскольку это легко испортить.

# Explicitly create a function for `.f`.
# This requires using the variable names (x, y, z) three times.
# It's completely clear what it's doing, but needs a lot of typing.
# It might sometimes fail - see https://github.com/tidyverse/purrr/issues/280

df_explicit <- df %>%
  mutate(
    avg = x - mean(x),
    a = pmap(.l = list(x, y, z), .f = function(x, y, z){ c(sum(x), sum(y), sum(z)) })
  )

# Pass the whole of `df` to `.l` and add `...` in an explicit function to deal with any unused columns. 
# variable names are used twice.
# `df` will have to be passes explicitly if not using pipes (eg, `mutate(.data = df, a = pmap(.l = df, ...`).
# This is probably inefficient for large datasets.

df_dots <- df %>%
  mutate(
    avg = x - mean(x),
    a = pmap(.l = ., .f = function(x, y, z, ...){ c(sum(x), sum(y), sum(z)) })
  )

# Use `pryr::f` (as discussed in https://stackoverflow.com/a/51123520/4269699).
# Variable names are used twice.
# Potentially unexpected behaviour.
# Not obvious to the casual reader why the extra `pryr::f` is needed and what it's doing

df_pryrf <- df %>%
  mutate(
    avg = x - mean(x),
    a = pmap(.l = list(x,y,z), .f = pryr::f({c(sum(x), sum(y), sum(z))} ))
  )

# Use `rowwise()` similar to this: https://stackoverflow.com/a/47734073/4269699
# Variable names are used once.
# It will mess up any vectorised functions used elsewhere in mutate, hence the two `mutate()`s

df_rowwise <- df %>%
  mutate( avg = x - mean(x) ) %>%
  rowwise() %>%
  mutate( a = list( {c(sum(x), sum(y), sum(z))} ) ) %>%
  ungroup()

# Use Romain Francois' neat {rap} package.
# Variable names used once.
# Like `rowwise()` it will mess up any vectorised functions so it needs two `mutate()`s for this particular problem
#

library('rap') #devtools::install_github("romainfrancois/rap")
df_rap <- df %>%
  mutate( avg = x - mean(x) ) %>%
  rap( a = ~ c(sum(x), sum(y), sum(z)) )

# Another solution discussed here https://stackoverflow.com/a/51123520/4269699 doesn't seem to work inside `mutate()`, but maybe could be tweaked?
# Like the `pryr::f` solution, it's not immediately obvious what the purpose of the `with(list(...` bit is.

df_with <- df %>%
  mutate(
    avg = x-mean(x),
    a = pmap(.l = list(x,y,z), .f = ~with(list(...), { c(sum(x), sum(y), sum(z))} ))
  )

Насколько я знаю, это варианты, исключая сопоставление позиций,

В идеале было бы возможно что-то подобное следующему, где функция qmap знает, как найти (в ряд) переменные x, y и z из объекта, переданного в mutate s.data аргумент.

df_new <- df %>%
  mutate(
    avg = x-mean(x),
    a = qmap( ~c(sum(x), sum(y), sum(z)) )
  )

Но я не знаю, как это сделать, поэтому рассмотрим это только частичный ответ.

Вопросы, связанные с данной:

...