Замена значений в столбцах в кадре данных на месте - PullRequest
3 голосов
/ 11 апреля 2019

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

Вот пример:

library(tidyverse)

Kid = c('Alfie', 'Brenda', 'Charlie', 'Dalma')
Likes_pie = c('Yes', 'Yes', 'No', NA)
Likes_sunshine = c(NA, 'Yes', 'Yes', 'No')
Likes_friendship = c('Yes', NA, 'Yes', 'No')
my.d = cbind(Kid, Likes_pie, Likes_sunshine) %>% as_tibble()

Я хочу поменять местамиОт «Да» до «Т» и от «Нет» до «F» в Likes_pie и Likes_sunshine.Но я хочу указать эти два столбца и пропустить Likes_friendship.Я хочу сохранить NA-ы.Нет никаких значений, кроме «Да» и «Нет».

Можно сделать с одним столбцом:

my.d = my.d %>% 
  mutate(
    Likes_pie = case_when(
      Likes_pie == 'Yes' ~ 'T',
      Likes_pie == 'No' ~ 'F'
    )
  )

Но это не работает в функции:

valConverter = function(d, var.value){
  d = d %>% 
    mutate(
      var.value = case_when(
        var.value == 'Yes' ~ 'T',
        var.value == 'No' ~ 'F'
      )
    )
return(d)
}

my.d = valConverter(my.d, Likes_sunshine)
my.d$Likes_pie
my.d$Likes_sunshine # :E

Предполагая, что это так, как бы я заменил значения столбцов для каждого столбца?Sg как mutate_if(d, c('Likes_pie', 'Likes_sunshine'), function(x) nomConverter(x)) путь?

Ответы [ 4 ]

4 голосов
/ 11 апреля 2019

Решение с использованием lapply и ifelse. Я предполагаю, что Likes_friendship является одним из столбцов (Иначе, почему вы создали этот вектор?), Но то, находится ли столбец во фрейме данных, не влияет на эффективность этого решения.

Кстати, T и F - зарезервированное слово в R, обозначающее логические значения TRUE и FALSE соответственно. Символ "T" и "F" может сбить с толку других.

# Store the column name you want to change
cols <- c("Likes_pie", "Likes_sunshine")

my.d[cols] <- lapply(my.d[cols], function(x) ifelse(x == "Yes", "T", "F"))
my.d
# # A tibble: 4 x 4
#   Kid     Likes_pie Likes_sunshine Likes_friendship
#   <chr>   <chr>     <chr>          <chr>           
# 1 Alfie   T         NA             Yes             
# 2 Brenda  T         T              NA              
# 3 Charlie F         T              Yes             
# 4 Dalma   NA        F              No     

DATA

library(tidyverse)

Kid = c('Alfie', 'Brenda', 'Charlie', 'Dalma')
Likes_pie = c('Yes', 'Yes', 'No', NA)
Likes_sunshine = c(NA, 'Yes', 'Yes', 'No')
Likes_friendship = c('Yes', NA, 'Yes', 'No')
my.d = cbind(Kid, Likes_pie, Likes_sunshine, Likes_friendship) %>% as_tibble()
2 голосов
/ 11 апреля 2019

Если целью является преобразование в логический столбец, просто выполните ==

library(dplyr)
my.d %>% 
   mutate_at(vars(cols),  list(~ . == "Yes"))

, где

cols <- c("Likes_pie", "Likes_sunshine")
2 голосов
/ 11 апреля 2019

Вы можете использовать mutate_at(), чтобы сделать это одним ударом, ifelse будет делать то, что вы хотите:

(my.d <- tibble(Kid, Likes_pie, Likes_sunshine, Likes_friendship) %>%
  mutate_at(c("Likes_pie", "Likes_sunshine"), list(~ifelse(. == "Yes", T, F))))

# A tibble: 4 x 4
  Kid     Likes_pie Likes_sunshine Likes_friendship
  <chr>   <lgl>     <lgl>          <chr>           
1 Alfie   TRUE      NA             Yes             
2 Brenda  TRUE      TRUE           NA              
3 Charlie FALSE     TRUE           Yes             
4 Dalma   NA        FALSE          No 
2 голосов
/ 11 апреля 2019

Не могли бы вы просто использовать mutate_all?

my.d %>%
    mutate_all(~case_when(
        .x == "Yes" ~ "T",
        .x == "No" ~ "F",
        TRUE ~ .x))
## A tibble: 4 x 3
#  Kid     Likes_pie Likes_sunshine
#  <chr>   <chr>     <chr>
#1 Alfie   T         NA
#2 Brenda  T         T
#3 Charlie F         T
#4 Dalma   NA        F

Или для определенных столбцов используйте mutate_at

cols <- c("Likes_pie", "Likes_sunshine")
my.d %>%
    mutate_at(vars(cols), ~case_when(
        .x == "Yes" ~ "T",
        .x == "No" ~ "F",
        TRUE ~ .x))

дает тот же результат.

...