复数组上带因子的Rowsum

Question

我有一个很长的复杂数组，我想在上面做

rowsum(my_array, group = factors)

不幸的是，rowsum不接受复数，而rowSums接受，但不接受因子，以求行的平均值。

你会推荐什么？

编辑:示例

my_array <- array(c(1+1i, -1-1i, 2, 0, 1+1i, 2+2i), dim = c(3, 2))
factors <- c(1, 1, 2)

预期响应：

     [,1] [,2]
[1,]    0 1+1i
[2,]    2 2+2i

第二次编辑：

因子分别应用于每一列，此处为0 == 1+1i + (-1-1i)和1+1i == 0 + 1+1i。

Answer 1

my_array是

    [,1] [,2]
[1,]  1+1i 0+0i
[2,] -1-1i 1+1i
[3,]  2+0i 2+2i

从预期的响应和第二个编辑中可以清楚地看出，factors用于将数组拆分为以下两个数组

 $`1`
   [,1] [,2]
   [1,]  1+1i 0+0i
   [2,] -1-1i 1+1i

   $`2`
   [,1] [,2]
   [1,] 2+0i 2+2i

然后对每个阵列的列求和。在base R中实现这一点的一种方法如下。使用by将数组拆分为按factors分组的数组列表，然后使用colSums对每个数组的列分别求和。之所以使用colSums，是因为它可以处理复数。

y <- by(my_array, factors, FUN = colSums)

这给出了有点冗长但清晰的结果：

  INDICES: 1
      V1   V2 
    0+0i 1+1i
    -------------------------------------------------------------------------------------------------------------- 
    INDICES: 2
      V1   V2 
    2+0i 2+2i

其中V1和V2是每个因子的数组列的总和。最后，我们将使用以下命令将其放入数组中

t(simplify2array(y))

这给了我们

    V1   V2
1 0+0i 1+1i
2 2+0i 2+2i

这是预期的反应。