Início / coding / Perguntas / 78548159
Accepted
Maël
Asked: 2024-05-29 16:08:14 +0800 CST

Nome dinâmico com cola na chamada mutada

  • 772

Quero criar uma função que tome como primeiro argumento o nome de um conjunto de dados e, como segundo argumento, parte do nome de uma coluna do dataframe. Quero então usar gluepara construir dinamicamente o nome da coluna, na função, e usar essa coluna em uma mutatechamada, assim:

library(tidyverse)

tmp <- 
  function(data, type){
  var <- glue::glue("Sepal.{type}")
  iris |> 
    select({{var}}) |> 
    mutate("{var}" := mean({{var}}))
}

Eu tentei muitas coisas, mas tenho dificuldade em encontrar uma solução em que a coluna seja chamada tanto pelo nome da nova coluna (aqui, "{var}") quanto pelo cálculo da nova coluna (aqui, mean({{var}})). O que se deve fazer nesses casos?

Aqui, a chamada tmp(iris, "Length")deve retornar um 150x1data.frame com o valor médio em todas as linhas.

tidyversesolução são preferidas ou quaisquer respostas baseadas em pipe.

3 respostas

  1. Best Answer

    Você pode usar mean({{var}})se modificar um pouco seu código, por exemplo, usando as.symbolto define var, em vez de um gluechar

    tmp <- function(data, type) {
    var <- as.symbol(glue::glue("Sepal.{type}"))
    data |>
        select(var) |>
        mutate("{var}" := mean({{ var }}))
}

    Para algumas alternativas, acho que você pode tentar get(var)ou !!rlang::syms(var), por exemplo

    tmp <- function(data, type) {
    var <- glue::glue("Sepal.{type}")
    data |>
        select({{ var }}) |>
        mutate("{var}" := mean(get(var)))
}

    ou

    tmp <- function(data, type) {
    var <- rlang::sym(glue::glue("Sepal.{type}"))
    data |>
        select(var) |>
        mutate("{var}" := mean(!!var))
}
    • 7

  2. Aqui está uma tentativa de pipe usando base, pena que não consigo encontrar uma maneira de usar o nome da variável dinâmica dentro de transform .

    tmpBase <- function(data, type){
  var <- paste0( "Sepal.", type)
  data |>
  transform(x = mean(get(var))) |>
    subset(select = x) |>
    setNames(var)
  }

head(tmpBase(iris, "Length"))
#   Sepal.Length
# 1     5.843333
# 2     5.843333
# 3     5.843333
# 4     5.843333
# 5     5.843333
# 6     5.843333
    • 4

  3. É isso que você está procurando?

    tmp <- 
  function(data, type){
  var <- glue::glue("Sepal.{type}")
  data |> 
    select(all_of(var)) |> 
    mutate(across(everything(), mean))
}

    Para select()variáveis ​​usando um vetor de caracteres, basta usar all_of(). E se você estiver selecionando apenas a variável a ser transformada, poderá usar mutate(across(everything(), f)para evitar a especificação do nome da variável.

    • 2

relate perguntas