[R] Subsetting an array by a vector of dimensions

Richard Pearson richard.pearson at postgrad.manchester.ac.uk
Fri Jul 11 18:30:03 CEST 2008


My understanding of matrix subscripting is that this can be used to access arbitrary elements from an array and return them as a vector, but I don't understand how that helps me here. I've now written a function that seems to do what I originally wanted, but I've also realised I want to do assignment too. I need to read up more on writing assignment functions to do this. However, I'm thinking that someone has already solved this, in a far more elegant way than my hacking :-). Here's my function for anyone interested:

> subsetArray <- function(x, subset) {
+   subsetString <- paste(subset, collapse=",")
+   subsetString <- gsub("NA","",subsetString)
+   evalString <- paste(expression(x), "[", subsetString, "]")
+   eval(parse(text=evalString))
+ }
> x <- array(1:24, dim=2:4)
> subsetArray(x, c(1,1,2))
[1] 7
> subsetArray(x, c(1,NA,2))
[1]  7  9 11
> subsetArray(x, c(1,NA,NA))
     [,1] [,2] [,3] [,4]
[1,]    1    7   13   19
[2,]    3    9   15   21
[3,]    5   11   17   23
> 
> subsetArray(x, c(1,1,2)) <- 25
Error in subsetArray(x, c(1, 1, 2)) <- 25 : 
  could not find function "subsetArray<-"

Best wishes

Richard.


Patrick Burns wrote:
> I think you are looking for subscripting with a
> matrix:
> 
> x[cbind(1,1,2)]
> 
> See, for instance, the subscripting section of chapter 1
> of S Poetry.
> 
> 
> Patrick Burns
> patrick at burns-stat.com
> +44 (0)20 8525 0696
> http://www.burns-stat.com
> (home of S Poetry and "A Guide for the Unwilling S User")
> 
> Richard Pearson wrote:
>> Hi
>>
>> Is it possible to subset an n-dimensional array by a vector of n 
>> dimensions? E.g. assume I have
>>
>>> x <- array(1:24, dim=2:4)
>>> x[1,1,2]
>> [1] 7
>>> dims <- c(1,1,2)
>>
>> I would like a function that I can supply x and dims as parameters to, 
>> and have it return 7. Also, I would like to do something like:
>>
>>> x[1,1,]
>> [1]  1  7 13 19
>>> dims2<- c(1,1,NA)
>>
>> And have a function of x and dims2 that gives me back [1]  1  7 13 19
>>
>> Does such a thing exist?
>>
>> Thanks
>>
>> Richard
>>
>>
> 

-- 
Richard D. Pearson             richard.pearson at postgrad.manchester.ac.uk
School of Computer Science,    http://www.cs.man.ac.uk/~pearsonr
University of Manchester,      Tel: +44 161 275 6178
Oxford Road,                   Mob: +44 7971 221181
Manchester M13 9PL, UK.        Fax: +44 161 275 6204



More information about the R-help mailing list