具有多個列和NA的R嵌套Ifelse

Question

我有一個用於比賽的數據框：

df <- data.frame(ID=c(1,2,3,4,5,6), Condition=c(1,2,2,1,1,2), White=c(1,1,NA,NA,NA,NA), Black=c(2,NA,NA,NA,2,NA), Asian=c(NA, NA, NA, 3, 3, 3), AmerIndian=c(NA,NA,4,NA,NA,NA), NatHawaiian=c(NA, NA, NA, 5, NA, NA))

並且我想開發一個新的比賽領域，無論條件2為何，都將填寫一個新領域。

這是我嘗試的：

df$var <-ifelse(as.numeric(df$White)==1&!is.na(df$White),"White", 
+     ifelse(as.numeric(df$Black)==2&!is.na(df$Black),"Black",
+        ifelse(as.numeric(df$Asian)==3&!is.na(df$Asian),"Asian",
+           ifelse(as.numeric(df$AmerIndian)==4&!is.na(df$AmerIndian),"AmerIndian",
+               ifelse(as.numeric(df$NatHawaiian)==5&!is.na(df$NatHawaiian),"NatHawaiian",NA)))))

我收到此錯誤：

Error in +ifelse(as.numeric(df$NatHawaiian) == 5 & !is.na(df$NatHawaiian),  : 
  invalid argument to unary operator

Answer 1

根據您輸入的代碼和生成的錯誤，我建議在每個ifelse語句的開頭刪除加號。 完成后，您共享的代碼：

df$var <- ifelse(
  as.numeric(df$White) == 1 & !is.na(df$White),
  "White", ifelse(
    as.numeric(df$Black) == 2 & !is.na(df$Black),
    "Black", ifelse(
      as.numeric(df$Asian) == 3 & !is.na(df$Asian),
      "Asian", ifelse(
        as.numeric(df$AmerIndian) == 4 & !is.na(df$AmerIndian),
        "AmerIndian", ifelse(
          as.numeric(df$NatHawaiian) == 5 &
            !is.na(df$NatHawaiian),
          "NatHawaiian",
          NA
        )
      )
    )
  )
)

將生成此輸出：

  ID Condition White Black Asian AmerIndian NatHawaiian        var
1  1         1     1     2    NA         NA          NA      White
2  2         2     1    NA    NA         NA          NA      White
3  3         2    NA    NA    NA          4          NA AmerIndian
4  4         1    NA    NA     3         NA           5      Asian
5  5         1    NA     2     3         NA          NA      Black
6  6         2    NA    NA     3         NA          NA      Asian

Answer 2

您可以嘗試使用tidyverse例如：

df %>%
  gather(key = "race", value = "val", 3:7) %>%
  mutate( rc = if_else((Condition == 2 & !is.na(val)), race, NULL)) %>%
  spread(race, val)