5

R で多次元配列をサブセット化する関数が必要です。キャッチは、関数が呼び出されるまで、どの次元またはその次元に沿った長さがわからないことです。

# subset a 3-d array; leave dims 1 and 2, but start 3rd dim at its 11th value
mydim <- dim(myarr)
myarr[, , 11:mydim[3]]

# subset a 4-d array; leave dims 1, 3 and 4, but start 2rd dim at its 8th value
mydim <- dim(myarr)
myarr[, 8:mydim[2], , ]

私は常に正確に 1 つの次元に沿ってサブセット化する必要があり、サブセット化は常に 1 以外の値から開始する必要があります。また、配列構造を保持する必要があるため、配列へのマトリックス インデックス付けは魅力的に見えません。前もって感謝します。

4

2 に答える 2

1

行列に基づいて配列をサブセット化する可能性を利用するオプションを次に示します。

myarr <- array(1:(2*3*4), dim = c(2, 3, 4))

myfun <- function(arr, from, len, Dim){
    dimArr <- dim(arr)
    if(missing(len)){
        subIdx <- from:dimArr[Dim]
    } else {
        subIdx <- from:(from + len - 1)
    }
    arrD <- lapply(as.list(dimArr), seq_len)
    arrD[[Dim]] <- subIdx
    subMat <- as.matrix(do.call(expand.grid, arrD))
    array(arr[subMat], dim = lapply(arrD, length))
}

> myfun(myarr, 2, 1, 3)
, , 1

     [,1] [,2] [,3]
[1,]    7    9   11
[2,]    8   10   12

> myfun(myarr, 2, Dim = 3)
, , 1

     [,1] [,2] [,3]
[1,]    7    9   11
[2,]    8   10   12

, , 2

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

, , 3

     [,1] [,2] [,3]
[1,]   19   21   23
[2,]   20   22   24
于 2012-11-28T21:13:35.483 に答える
1

どうぞ！あまりにもテストされた...

  sampleArray <- function(myarr, dm, start, leng) {
  ## arguments: 
  ##  dm is the dimension selected
  ##  start is where in dm to being
  ##  leng is how far in from dm to go  
  ##   start+leng <= dim(myarr)[dm]

    leng <- leng-1

    # error check
    if (start+leng > dim(myarr)[dm])
      warning("leng too long by ", start+leng - dim(myarr)[dm], ".")

    #initialize a vector of all TRUE
    indx <- as.list(rep(TRUE, length(dim(myarr))))

    # change the required dimension to the required sequence
    indx[[dm]] <- seq(start, start+leng)
    indx <- paste(indx, collapse=",")

    expr <- paste0("myarr[", indx, "]")

    # return the appropriate sample
    eval(parse(text=expr))
  }

例：

# sample array
myarr <- array(1:2250, dim=c(15, 10, 15))

# example call
sampleArray(myarr, dm=2, start=4, leng=3)
于 2012-11-28T21:10:00.207 に答える