时间戳

首页 > 解决方案 > 如果使用 case_when，则对两个或更多列进行变异

r - 如果使用 case_when，则对两个或更多列进行变异

问题描述

我正在尝试将case_when函数用于一堆列 ya data.frame。

这种情况下不返回指定的列mutate

cars %>% mutate (
  km = speed * dist,
  mt = km / 1000
) %>%
mutate (
  .funs = case_when(
    (speed < 20 ) ~ {
      km = km * 2 
      mt = mt * 3
    }
  )
)

谢谢

标签： rdplyr

解决方案

我们可以使用mutate_at

library(tidyverse)
cars %>%
   mutate(km = speed * dist, mt = km/1000) %>%
   mutate_at(vars(km, mt), funs(case_when(speed < 20 ~ .*2,
                                      TRUE ~ .)))

如果我们需要为每一列使用单独的值进行计算，则使用map2或pmap

out <- cars %>%
         mutate(km = speed * dist, mt = km/1000)  %>%  
         select(km, mt) %>%
         map2_df(., list(2, 3), ~ 
           case_when(cars$speed < 20 ~ .x * .y, TRUE ~ .x)) %>% 
         bind_cols(cars, .)

head(out)
#  speed dist  km    mt
#1     4    2  16 0.024
#2     4   10  80 0.120
#3     7    4  56 0.084
#4     7   22 308 0.462
#5     8   16 256 0.384
#6     9   10 180 0.270

推荐阅读