r - colSums() 的特殊用途，仅当缺少 1 个或更少时 na.rm = TRUE

Question

我需要对 data.frame 中的一些列求和，规则是，如果缺少一个以上的观察值，则将列求和为 NA NA如果仅缺少 1 个或更少，则无论如何都要对其求和。

假设我有这样的数据，

dfn <- data.frame(
a  = c(3, 3, 0, 3),
b  = c(1, NA, 0, NA),
c  = c(0, 3, NA, 1))

dfn
  a  b  c
1 3  1  0
2 3 NA  3
3 0  0 NA
4 3 NA  1

然后我应用我的规则，并对少于 2 个缺失 NA 的列求和。所以我得到了这样的东西。

  a  b  c
1 3  1  0
2 3 NA  3
3 0  0 NA
4 3 NA  1
5 9 NA  4

我玩过 colSums(dfn, na.rm = FALSE) 和 colSums(dfn, na.rm = TRUE)。在我的真实数据中，有超过三列，也有超过 4 行。我想我可以通过某种方式计算缺失的数量并将其用作规则？

Answer 1

我不认为你可以单独使用 colSums 来做到这一点，但你可以使用 ifelse 添加到它的结果:

colSums(dfn,na.rm=TRUE) + ifelse(colSums(is.na(dfn)) > 1, NA, 0)
 a  b  c 
 9 NA  4