如何让 switch() 处理 NA？答案

【问题标题】：How to get switch() to handle NA?如何让 switch() 处理 NA？
【发布时间】：2020-09-28 14:17:05
【问题描述】：

好的，我必须重新编码一个 df，因为我希望因子是整数：

library(dplyr)

load(url('http://biostat.mc.vanderbilt.edu/wiki/pub/Main/DataSets/crash2.rda'))

df <- crash2 %>% select(source, sex)

df$source <- sapply(df$source, switch, "telephone" = 1, "telephone entered manually" = 2, "electronic CRF by email" = 3, "paper CRF enteredd in electronic CRF" = 4, "electronic CRF" = 5, NA)

这按预期工作，但下一个变量（性别）中有 NA，事情变得复杂：

df$sex <- sapply(df$sex, switch, "male" = 1, "female" = 2, NA)

返回一个列表，其中 NA 被切换到遗忘状态。使用 unlist() 返回一个对于 df 来说太短的向量。

length(unlist(sapply(df$sex, switch, "male" = 1, "female" = 2, NA)))

应该是20207，但是是20206。

我想要的是通过将 NA 作为 NA 返回来匹配 df 的向量。

除了一个可行的解决方案之外，我会非常感谢您解释我出错的地方以及代码的实际工作原理。

编辑：感谢您的所有回答。通常情况下，我应该自己注意到一个更有效的解决方案（好吧，我自己注意到了，但显然为时已晚）：

>str(df$sex)
Factor w/ 2 levels "male","female": 1 2 1 1 2 1 1 1 1 1 ...

所以我可以使用as.numeric() 来获得我想要的东西。

【问题讨论】：

可能不需要循环或切换；试试x = c("a", "e", "a", "a", NA, "d", "b", "b", NA, "d"); lu = c( "a"=1, "b"=2, "c"=3, "d"=4, "e"=5); lu[x]

标签： r sapply

【解决方案1】：

您可以使用`NA`。

x
# [1] "a" "e" "a" "a" NA  "d" "b" "b" NA  "d"
unname(sapply(x, switch, "a"=1, "b"=2, "c"=3, "d"=4, "e"=5, `NA`=NA))
# [1]  1  5  1  1 NA  4  2  2 NA  4

数据：

x <- c("a", "e", "a", "a", NA, "d", "b", "b", NA, "d")

【讨论】：

什么是``NA``在做什么？
@YOLO 你也可以写"NA"（如果你有"NA"作为x中的字符，则不是），它将NA标识为另一个级别（如字符"NA" ) 和"NA"=NA 将NA 分配给它。
switch(NA_character_, NA= 1) 是 1 ，而 switch(NA, NA= 1) 是 NULL，因为第一个参数不是强制为字符而是整数，真的不是我的预计。

【解决方案2】：

如果您有兴趣，还可以通过dplyr 使用case_when() 进行此操作：

load(url('http://biostat.mc.vanderbilt.edu/wiki/pub/Main/DataSets/crash2.rda'))

df <- crash2 %>% dplyr::select(source, sex) %>% 
  mutate(source = case_when(
    source == "telephone"~1, 
    source == "telephone entered manually"~2, 
    source == "electronic CRF by email"~3, 
    source == "paper CRF enteredd in electronic CRF"~4, 
    source == "electronic CRF"~5), 
    sex = case_when(
      sex == "male" ~ 1, 
      sex == "female" ~ 2))

table(df$sex, useNA="ifany")
#     1     2  <NA> 
# 16935  3271     1

【讨论】：