我想我已经对这些评论获得了非常好的回答,但是我将重新表述该问题以供将来引用。
我正在尝试使用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)
}
最佳答案
根据其他用户的建议,我将发布问题的答案。该解决方案由@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_,因为我一直在获取有关类型的错误。
关于r - 仅对非NA元素求和,但如果所有NA都返回NA,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/41470277/