时间戳

首页 > 解决方案 > 使用 purrr map2 将嵌套数据中的 if then 构造转换为 case_when

r - 使用 purrr map2 将嵌套数据中的 if then 构造转换为 case_when

问题描述

这是我的数据：

vec1 <- c("A", "B")
vec2 <- c("A", "B", "C")

df1 <- structure(list(A = c("X", "Y"), data = list(structure(list(B = c(4L, 9L)), .Names = "B", row.names = c(NA, -2L), class = c("tbl_df", "tbl", "data.frame")), structure(list(B = c(5L, 2L, 8L, 2L)), .Names = "B", row.names = c(NA, -4L), class = c("tbl_df", "tbl", "data.frame"))), C = list(c("A", "B"), c("A", "B", "C"))), class = c("tbl_df", "tbl", "data.frame"), row.names = c(NA, -2L), .Names = c("A", "data", "C"))

这是我的代码：

df2 <- df1 %>% mutate(data = map2(data, C, ~ if(identical(.y, vec1)) filter(.x, B < 5) else filter(.x, B > 4)))

为了轻松扩展条件的数量，我想将上面代码中的 if else 构造转换为 case_when 构造。我怎样才能做到这一点？

标签： rdplyrtidyversepurrr

解决方案

为什么需要一个case when？

尝试：

m=df1%>%
  mutate(filt=map2(data,match(C,list(vec1,vec2)),
               ~filter_(.x,c("B<=4","B>4")[.y])))

   m
# A tibble: 2 x 4
  A     data             C         filt            
  <chr> <list>           <list>    <list>          
1 X     <tibble [2 x 1]> <chr [2]> <tibble [1 x 1]>
2 Y     <tibble [4 x 1]> <chr [3]> <tibble [2 x 1]>

如果您希望它与您df2刚刚替换的data列相同：

df1%>%
      mutate(data=map2(data,match(C,list(vec1,vec2)),
                   ~filter_(.x,c("B<=4","B>4")[.y])))%>%
      identical(df2)
 [1] TRUE

如果条件太多，你可以在外面创建它们。条件的顺序确实很重要：顺序或条件应与向量的顺序相匹配，例如：

   conditions=c("B>4","B<=4")
   ll=list(vec2,vec1)
   df1%>%
      mutate(data=map2(data,match(C,ll),
                   ~filter_(.x,conditions[.y])))%>%
      identical(df2)
     [1] TRUE

推荐阅读