r - 如果所有汇总值均为 NA，则 dplyr 汇总保留 NA

Question

我想使用 dplyr 汇总按组汇总计数。具体来说，如果不是所有求和值都是 NA，我想删除 NA 值，但如果所有求和值都是 NA，我想显示 NA。例如:

name <- c("jack", "jack", "mary", "mary", "ellen", "ellen")
number <- c(1,2,1,NA,NA,NA)

df <- data.frame(name,number)

在这种情况下，我想要以下结果:

jack = 3

玛丽 = 1

艾伦 = NA

但是，如果我设置 na.rm = F :

df %>% group_by(name) %>% summarise(number = sum(number, na.rm = F))

结果是:

jack = 3

玛丽 = NA

艾伦 = NA

如果我设置 na.rm = T :

df %>% group_by(name) %>% summarise(number = sum(number, na.rm = T))

结果是

jack = 3

玛丽 = 1

艾伦 = 0

我怎样才能解决这个问题，以便带有数字和 NA 的案例得到一个数字作为输出，但只有 NA 的案例得到 NA 作为输出。

Answer 1

我们可以有一个 if/else条件 - if all 'number 中的值为 NA ，然后返回 NA或 else获取 sum

library(dplyr)
df %>% 
  group_by(name) %>% 
  summarise(number = if(all(is.na(number))) NA_real_ else sum(number, na.rm = TRUE))