如何使用 tapply 并保留值的顺序

Question

希望这不是一个太愚蠢的问题，但作为一个 R 初学者，我对 tapply 有一个严重的问题。可以说

factors <- as.factor( c("a", "b", "c", "a", "b", "c", "a", "b", "c") )
values  <- c( 1, 2, 3, 4, 5, NA, 7, NA, NA )
tapply(
  values,
  factors,
  function(x){
    if( sum(is.na(x)) == 1 ){
      x[ is.na(x) ] <- 0
    }
    return(x)
  }
)

结果是

$a
[1] 1 4 7

$b
[1] 2 5 0

$c
[1]  3 NA NA

但是，我需要的是返回一个保留值的原始顺序的向量，即：

c( 1,2,3,4,5,NA,7,0,NA )

提前非常感谢。

Answer 1

在这种情况下，您应该使用 ave 函数：

> ave(values, factors, FUN=function(x) {
+     if( sum(is.na(x)) == 1 ){
+       x[ is.na(x) ] <- 0
+     }
+     return(x)
+   }
+ )
[1]  1  2  3  4  5 NA  7  0 NA

Answer 2

一个简单的 for 循环可以非常简单地完成此操作：

fun <- function(x){
   if(sum(is.na(x)) == 1){x[is.na(x)] <- 0}
       return(x)
}

for (i in unique(factors)){
   values[i == factors] <- fun(values[i == factors])
}

Answer 3

一种选择是使用 split() 的替换方法：

## create a copy to store the result after replacement
res <- values

## use split's replacement method to split, apply, and recombine
split(res, factors) <- lapply(split(res, factors),
 function(x){
 if( sum(is.na(x)) == 1 ){
   x[ is.na(x) ] <- 0
 }
  return(x)
 }
)

Answer 4

如果其他人通过搜索如何禁用组的字母排序发现此问题，您可以执行以下操作：

> v=1:4
> group=c("b","b","a","a")
> tapply(v,group,sum)
a b
7 3
> tapply(v,factor(group,unique(group)),sum)
b a
3 7