r - 仅对非NA元素求和，但如果所有NA都返回NA

Question

我想我已经对这些评论获得了非常好的回答，但是我将重新表述该问题以供将来引用。

我正在尝试使用data.table进行分组汇总。问题在于，某些组只有具有NA。对于这些组，我希望总和返回NA。但是，如果存在一组值与NA不同的组，我想获得非NA值的总和。

A <- data.table(col1= c('A','A','B','B','C','C'),  
                col2= c(NA,NA,2,3,NA,4))

这没有添加参数 na.rm = T，C组应返回4时返回NA。

A[, sum(col2), by = .(col1)]
   col1 V1
1:    A NA
2:    B  5
3:    C NA

但是，添加 na.rm = T时应返回NA时，在A组中返回0。

A[, sum(col2, na.rm = T), by = .(col1)]
   col1 V1
1:    A  0
2:    B  5
3:    C  4

我最喜欢的方法是Sandipan在评论中建议的方法，类似于我在下面编写的功能:

ifelse(all(is.na(col2)), NA, sum(col2, na.rm = T)

我创建了一个函数来解决它，但是我不确定是否已经有内置的方法来解决这个问题:

sum.na <- function(df){

  if (all(is.na(df))){

    suma <- NA
  }  
  else {    
    suma <- sum(df, na.rm = T)
  }

  return(suma)
}

Answer 1

根据其他用户的建议，我将发布问题的答案。该解决方案由@sandipan在上面的评论中提供:

如问题中所述，如果您需要对包含NA的一列的值求和，则有两种不错的方法:

1)使用ifelse:

A[, (ifelse(all(is.na(col2)), col2[NA_integer_], sum(col2, na.rm = T))), 
  by = .(col1)]

2)定义@Frank建议的功能:

suma = function(x) if (all(is.na(x))) x[NA_integer_] else sum(x, na.rm = TRUE)

A[, suma(col2), by = .(col1)]

请注意，正如@Frank所指出的，我添加了NA_integer_，因为我一直在获取有关类型的错误。