时间戳

首页 > 解决方案 > 具有复杂数组因子的行和

r - 具有复杂数组因子的行和

问题描述

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

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。

标签： r

解决方案

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

然后对每个数组的列求和。在基础 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

这是预期的反应。

推荐阅读