[R] array manipulation simple questions

Gabor Grothendieck ggrothendieck at gmail.com
Tue Feb 24 06:31:43 CET 2009


2009/2/23 Λεωνίδας Μπαντής <bleonidas25 at yahoo.gr>:
>
> Hi there,
>
> I am pretty new to R. Actually I started using it yesterday. I have two questions:
>
> 1.   Suppose I have a<-c(1:10)  (or a<-array(c(1:10),dim=c(1,10)))
>
> and I want to end up with vector b=[0 0 0 0 0 1 1 1 1 1]. i.e. I want to substitute alla elements that are <5 with 0 and >5 with 1.
>
> I did this by using a for loop. Is there another function to use so as to avoid the "for"?


(a>5)+0

# or

out <- as.numeric(a > 5)
if (is.matrix(a)) out <- matrix(a, 1)
out

>
>
> 2.  Suppose I have a<-c(1,1,2,2,3,3)  (or array again)
>
> And I want to place a "4,5" before every "2" and end up with a new "bigger" vector "b":
>
> b=[1 1 4 5 2 4 5 2 3 3]

# return c(4, 5, 2) or x
f <- function(x) if (x==2) c(4:5, x) else x

# apply f to each element and
# if a is a matrix make output a matrix too

out <- unlist(sapply(a, f))
if (is.matrix(a)) out <- matrix(out, nrow(a))
out

>
> Also I want to find where the 2's in array "a" (if it was an array) are located i.e. positions 3,4.

which(a == 2)




More information about the R-help mailing list