struttura di conservazione dell'indicizzazione di array

Question

Ho bisogno di una funzione che subisca un array multidimensionale in R; il problema è che non conosco quale dimensione o lunghezza lungo quella dimensione fino a quando non viene chiamata la funzione.

# 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], , ] 

sempre necessario sottoinsieme lungo esattamente una dimensione, e il subsetting è sempre per iniziare ad un certo valore diverso da 1. Inoltre devo preservare la struttura a matrice, in modo matrice indicizzazione in array non ha un aspetto attraente. Grazie in anticipo.

Answer 1

Ecco un'opzione che sfrutta la possibilità di sottoinsiemi una matrice sulla base di una matrice:

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 

Answer 2

Eccoci! Testato anche ...

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)) 
    } 

esempio:

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

# example call 
sampleArray(myarr, dm=2, start=4, leng=3)