[R] Subsetting an array by a vector of dimensions

Wolfgang Huber huber at ebi.ac.uk
Fri Jul 11 23:48:43 CEST 2008


Hi Richard,

what is wrong with Patrick's suggestion? I get

x <- array(1:24, dim=2:4)
x[rbind(c(1,1,2))]
## [1] 7
x[rbind(c(1,1,2))] <- 13
x[rbind(c(1,1,2))]
## [1] 13

And you could also do

   do.call("[", list(x,1,1,2))

These should be a bit quicker than the eval/parse constructs - see also 	
         library("fortunes")
	fortune(106)

Also note that your "subsetArray<-" function will afaIu each time copy 
(and then modify) the whole array, which may be quite inefficient. I 
believe that the "[<-" function has been optimized to avoid that in some 
cases. Try

  x=numeric(1e6); system.time({for (i in 1:1000) x[i]=12})

versus

  x=numeric(1e6); system.time({for (i in 1:1000) {y=x; y[i]=12}})

	Best wishes
	Wolfgang

----------------------------------------------------
Wolfgang Huber, EMBL-EBI, http://www.ebi.ac.uk/huber




Richard Pearson wrote:
> In case anyone's still interested, I now have (I think!) a complete 
> solution (thanks to a quick look at my new favourite document - S Poetry 
> :-)
> 
> subsetArray <- function(x, subset) {
>   subsetString <- paste(subset, collapse=",")
>   subsetString <- gsub("NA","",subsetString)
>   evalString <- paste(expression(x), "[", subsetString, "]")
>   eval(parse(text=evalString))
> }
> 
> "subsetArray<-"<- function(x, subset, value) {
>   subsetString <- paste(subset, collapse=",")
>   subsetString <- gsub("NA","",subsetString)
>   evalString <- paste(expression(x), "[", subsetString, "] <-", 
> expression(value))
>   eval(parse(text=evalString))
>   x
> }
> 
> x <- array(1:24, dim=2:4)
> subsetArray(x, c(1,1,2))
> subsetArray(x, c(1,1,2)) <- 25
> x
> 
> Thanks to Pat!
> 
> Richard
> 
> 
> Richard Pearson wrote:
>> 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 a 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



More information about the R-help mailing list