Como transformar uma coluna em um dataframe dinamicamente em R

Oct 13 2020

Eu quero alterar o valor de uma coluna dataframe/ tibbledefinida como uma variável ( col_name). Tentei !!col_namesem sucesso.

library(dplyr)
#> 
#> Attaching package: 'dplyr'
#> The following objects are masked from 'package:stats':
#> 
#>     filter, lag
#> The following objects are masked from 'package:base':
#> 
#>     intersect, setdiff, setequal, union
col_name <- "b" # the column to select

df <- tibble(a = c(1,2,3), b = c(2,4,6))

df %>%
  mutate(b = if_else((b == 6 & a == 3), 8, b))  # this works 
#> # A tibble: 3 x 2
#>       a     b
#>   <dbl> <dbl>
#> 1     1     2
#> 2     2     4
#> 3     3     8

# but this doesn't
df %>%
  mutate(!!col_name := if_else((!!col_name == 6 & a == 3), 8, !!col_name))
#> Error: Problem with `mutate()` input `b`.
#> x `false` must be a double vector, not a character vector.
#> i Input `b` is `if_else(("b" == 6 & a == 3), 8, "b")`.
Created on 2020-10-13 by the reprex package (v0.3.0)

Respostas

2 RonakShah Oct 13 2020 at 10:26

Para usar !!no RHS, você precisa primeiro converter col_namepara símbolo.

library(dplyr)
df %>% mutate(!!col_name := if_else(!!sym(col_name) == 6 & a == 3, 
                            8, !!sym(col_name)))

Outras alternativas incluem o uso de get:

df %>% mutate(!!col_name := if_else(get(col_name) == 6 & a == 3,
                             8, get(col_name)))

Ou sem qualquer uso de NSE .data:

df %>% mutate(!!col_name := if_else(.data[[col_name]] == 6 & a == 3, 
                            8, .data[[col_name]]))
3 zx8754 Oct 13 2020 at 10:57

Usando a base :

df[ df[, col_name ] == 6 & df$a == 3, col_name ] <- 8

df
#   a b
# 1 1 2
# 2 2 4
# 3 3 8

Nota: Sim, estou ciente de que a pergunta é sobre "arrumar" , isso é apenas para ilustrar porque para algumas tarefas simples soluções básicas são tão boas / melhores.