使用下面的示例数据,我试图基于三列变量(type、adict和sex)的值创建一个新变量“category”。

但我想合并 type 和 addict 分成一组 sex 在另一组。 所以我用 any 要从逻辑上得到一组逻辑向量,至少要有一个值为真,或者两个值都为真。

df <- data.frame(type = c(NA, "bad",NA), addict=c('visky','wine',NA),
                 sex=c(NA,'male',NA))


> df
  type addict  sex
1 <NA>  visky <NA>
2  bad   wine male
3 <NA>   <NA> <NA>

library(dplyr)


df%>%
  mutate(category=ifelse(any(is.na(type)&addict=="visky")&any(is.na(sex)),"categ1",
         ifelse(any(type=="bad"|addict=="wine")&any(!is.na(sex)),"categ2",
         ifelse(any(is.na(type)&is.na(addict))&any(is.na(sex)),"categ3",NA))))


       type addict  sex category
1 <NA>  visky <NA>   categ1
2  bad   wine male   categ1
3 <NA>   <NA> <NA>   categ1

从中可以看出 ifelse 循环工作不正常我不明白为什么?

预期产量

       type addict  sex category
1 <NA>  visky <NA>   categ1
2  bad   wine male   categ2
3 <NA>   <NA> <NA>   categ3

THX提前

用户定义函数的更新 category

还有一件事,如果我想在不使用case的情况下编写用户定义的函数,我可能会使用

categ <- function(type,addict,sex){ 

if (any(is.na(type)&addict=="visky"&is.na(sex))){ 
"categ1" 
} 
else{ 
NA 
} 

}

但这也是给予

df%>%
mutate(category=categ(type,addict,sex))

  type addict  sex category
1 <NA>  visky <NA>   categ1
2  bad   wine male   categ1
3 <NA>   <NA> <NA>   categ1