如何在R中使用因子行求和

发布于01月25日

我有这样的数据:

df <- data.frame(id = c("001", "002", "003", "004"),
                 banana = c("banana", NA, NA, NA),
                 apple = c(NA, "apple", "apple", NA),
                 orange = c("orange", NA, NA, NA),
                 mango = c(NA, NA, NA, "mango"))

 id banana apple orange mango
001 banana    NA orange    NA
002     NA apple     NA    NA
003     NA apple     NA    NA
004     NA    NA     NA mango

我想通过以下方式将变量合并到变量flavor中:

 id banana apple orange mango flavor
001 banana    NA orange    NA  mixed
002     NA apple     NA    NA  apple
003     NA apple     NA    NA  apple
004     NA    NA     NA mango  mango

我该怎么做？任何帮助是赞赏的，虽然一个概括dplyr解决方案将是伟大的.

编辑:我需要一个解决方案稳健的其他列在我的数据框中也.

推荐答案

如你所要求的，这是一个dplyr的解决方案.

library(dplyr)

df %>% 
  pivot_longer(-id) %>% 
  mutate(flavor = ifelse(sum(is.na(value)) == ncol(df) - 2, value[!is.na(value)], "mixed"), .by = id) %>% 
  pivot_wider() %>% 
  relocate(-flavor) # move the flavor column to the last

# A tibble: 4 × 6
  id    banana apple orange mango flavor
  <chr> <chr>  <chr> <chr>  <chr> <chr> 
1 001   banana NA    orange NA    mixed 
2 002   NA     apple NA     NA    apple 
3 003   NA     apple NA     NA    apple 
4 004   NA     NA    NA     mango mango