以程序确定的维数的数组切片



是否有一种方法可以在R中优雅地指定多维数组中的切片,从而事先不知道维数?

例如,对于一个5维数组,可以使用

来分割最后一个维度
fourdimslice <- fivedimarray[,,,,1]

但是我想把它编码成任意尺寸,比如

slice <- arbitrarydimarray(dim=5, index=1)

我不明白do.call('[', ...)是否会是一种方法。我也尝试过命名维度,但a[fifth=1]没有按我想要的方式解释。

我想像apply()这样的函数需要做类似的事情。

您可以使用asub,来自abind包。

# Sample data: a 5-dimensional array
x <- array(seq_len(prod(2:6)), dim=2:6)
library(abind)
str( asub(x, 1, 5) )
# int [1:2, 1:3, 1:4, 1:5] 1 2 3 4 5 6 7 8 9 10 ...
str( x[,,,,1] )
# int [1:2, 1:3, 1:4, 1:5] 1 2 3 4 5 6 7 8 9 10 ...
all( asub(x, 1, 5) == x[,,,,1] )
# [1] TRUE

如果您不想使用abind包,那么您可以使用基于slice.index的双行代码。我不知道哪个更有效率。

x <- array(seq_len(prod(2:6)), dim=2:6)
library(abind)
str( y <- asub(x, 1, 5) )

或:

z <- x[slice.index(x,5)==1]

这提取了正确的元素,但维度丢失了。要恢复它们:

dim(z) <- dim(x)[-5]

(显然可以将其包装在适当的函数中。)测试:

all.equal(y,z)  ## TRUE

下面是提取第一个切片的奇怪解决方案

your.array=array(runif(27),dim=c(3,3,3)) # this is to create an example of array from which you want a slice.
array(your.array[1:prod(dim(your.array)[-1])],dim(your.array)[-1])

相关内容

  • 没有找到相关文章

最新更新