10

假设您有一个包含行、列和层的 3 维数组:

A <- array (1:27, c(3,3,3))

想象一下你有一个函数,它接受一个矩阵作为输入并返回一个矩阵作为输出,比如t.

如何将函数应用于数组的每一层,返回与第一个大小相同的另一个数组?

我觉得我应该能够以apply某种方式做到这一点,但我不能。

奖励问题(如果您回答这个问题,我将非常感激):这样做是否更快,或者列出每个层矩阵及其lapply函数?

--

编辑:请不要认为这个问题得到了回答——下面的答案没有回答这个问题。

4

2 回答 2

5

您必须考虑要提取值的边距。

3rd您可以通过应用维度 1 和 2(需要更好的词的行和列)来转置每个维度矩阵

apply(A,1:2,t)
, , 1

     [,1] [,2] [,3]
[1,]    1    2    3
[2,]   10   11   12
[3,]   19   20   21

, , 2

     [,1] [,2] [,3]
[1,]    4    5    6
[2,]   13   14   15
[3,]   22   23   24

, , 3

     [,1] [,2] [,3]
[1,]    7    8    9
[2,]   16   17   18
[3,]   25   26   27

您还可以使用plyrand aaplywhich 可能会更直观地执行

library(plyr)

aaply(A,3,t)
, ,  = 1


X1   1  2  3
  1  1  4  7
  2 10 13 16
  3 19 22 25

, ,  = 2


X1   1  2  3
  1  2  5  8
  2 11 14 17
  3 20 23 26

, ,  = 3


X1   1  2  3
  1  3  6  9
  2 12 15 18
  3 21 24 27

至于哪个更快lapplyapply,我认为也许lapply会赢,但是您仍然需要考虑要从中提取矩阵的边距。

我通常发现在一个维度上思考要容易得多。如果地球是平的,一切都会更直接!

于 2012-12-10T23:39:32.590 回答
1

换位的另一种可能性是 function aperm

A <- array (1:27, c(3,3,3))
# aperm permute the dimensions in the given order
# here we want to permute dimension 1 and 2
# so the new order is 2-1-3 instead of 1-2-3
aperm(A, c(2,1,3)) 
    , , 1

     [,1] [,2] [,3]
[1,]    1    2    3
[2,]    4    5    6
[3,]    7    8    9

, , 2

     [,1] [,2] [,3]
[1,]   10   11   12
[2,]   13   14   15
[3,]   16   17   18

, , 3

     [,1] [,2] [,3]
[1,]   19   20   21
[2,]   22   23   24
[3,]   25   26   27

就将函数应用于每一层(比如你想将每一层乘以不同的值)而言,这是另一种可能性:

vec <- c(1,5,3)
lapply(seq_along(vec), function(x)A[,,x]*vec[x])
[[1]]
     [,1] [,2] [,3]
[1,]    1    4    7
[2,]    2    5    8
[3,]    3    6    9

[[2]]
     [,1] [,2] [,3]
[1,]   50   65   80
[2,]   55   70   85
[3,]   60   75   90

[[3]]
     [,1] [,2] [,3]
[1,]   57   66   75
[2,]   60   69   78
[3,]   63   72   81

但是正如您所看到的,它创建了一个列表,因此它需要使用 function 多一步simplify2array

simplify2array(lapply(seq_along(vec),function(x)A[,,x]*vec[x]))
, , 1

     [,1] [,2] [,3]
[1,]    1    4    7
[2,]    2    5    8
[3,]    3    6    9

, , 2

     [,1] [,2] [,3]
[1,]   50   65   80
[2,]   55   70   85
[3,]   60   75   90

, , 3

     [,1] [,2] [,3]
[1,]   57   66   75
[2,]   60   69   78
[3,]   63   72   81

或直接使用sapply

sapply(seq_along(vec), function(x)A[,,x]*vec[x], simplify="array")
, , 1

     [,1] [,2] [,3]
[1,]    1    4    7
[2,]    2    5    8
[3,]    3    6    9

, , 2

     [,1] [,2] [,3]
[1,]   50   65   80
[2,]   55   70   85
[3,]   60   75   90

, , 3

     [,1] [,2] [,3]
[1,]   57   66   75
[2,]   60   69   78
[3,]   63   72   81
于 2012-12-12T10:16:33.073 回答