在R中创建列表时对对象用if语句(dplyr的mutate生成交互变量)
解决dplyr中动态生成交互变量的问题
你遇到的核心问题是:当condition$x为NULL时,直接在list()里嵌入if语句会产生NULL元素,导致interaction()无法正常工作。下面提供几种可行的解决方案:
方法1:动态构建变量列表(最直观)
先初始化基础变量列表,再根据条件添加需要的变量,最后用do.call()调用interaction():
set.seed(1) x <- sample(1:2, 10, replace = T) y <- sample(1:2, 10, replace = T) z <- sample(1:2, 10, replace = T) df <- data.frame(x, y, z) condition <- list() condition$x <- NULL # 测试NULL场景,改成1即可激活x的筛选和交互 df %>% { if(!is.null(condition$x[1])) filter(., x %in% condition$x) else . } %>% mutate( # 初始化包含y、z的列表 vars_list = list(y, z), # 若condition$x存在,将x加入列表头部 vars_list = if(!is.null(condition$x[1])) c(list(x), vars_list) else vars_list, # 生成交互变量 condition = do.call(interaction, vars_list) ) %>% select(-vars_list) # 移除临时列
方法2:用c()组合动态元素(更简洁)
利用c()会自动忽略NULL的特性,直接在interaction()里构建列表:
df %>% { if(!is.null(condition$x[1])) filter(., x %in% condition$x) else . } %>% mutate(condition = interaction( c( # 条件满足时返回list(x),否则返回NULL(会被c()忽略) if(!is.null(condition$x[1])) list(x), list(y, z) ) ))
方法3:用purrr::compact过滤NULL元素
如果已经在使用tidyverse,可借助purrr的compact()函数移除列表中的NULL元素:
library(purrr) df %>% { if(!is.null(condition$x[1])) filter(., x %in% condition$x) else . } %>% mutate(condition = interaction( compact(list( if(!is.null(condition$x[1])) x, y, z )) ))
为什么原代码会出错?
原代码中,当condition$x为NULL时,{if(!is.null(condition$x[1])) x}会返回NULL,此时list()会包含一个NULL元素:
list(NULL, y, z)
而interaction()无法处理包含NULL的输入列表,因此抛出错误。上面的方法都是通过移除或避免生成NULL元素来解决这个问题。
内容的提问来源于stack exchange,提问作者John
相关产品推荐
相关产品推荐

