2017-03-27 2 views
5

Ainsi, si l'entrée est un vecteur tel queComment inverser des nombres dans un vecteur SEULEMENT s'ils sont séquentiels?

v <- (1, 2, 3, 7, 2, 8, 9) 

La sortie serait

(3, 2, 1, 7, 2, 9, 8) 

J'ai essayé d'utiliser imbriqué et si des boucles avec la condition d'une fonction is.sorted, mais pas eu du succès.

+1

Voir aussi: [Comment partitionner un vecteur en groupes de réguliers, des séquences consécutives?] (Http: //stackoverflow.com/questions/5222061/how-to-partition-a-vec tor-into-groups-of-regular-consécutives-sequences) – Henrik

+0

... ce que je crois être un bon post canonique pour le 'cumsum (... diff (...' idiome. – Henrik

Répondre

8

Avec les données de l'échantillon

x <- c(1, 2, 3, 7, 2, 8, 9) 

Vous pouvez diviser en « groupes séquentiels » avec

grp <- cumsum(!c(TRUE, diff(x)==1)) 
grp 
# [1] 0 0 0 1 2 3 3 

Fondamentalement, nous regardons la diff de l'élément précédent et le suivi des changements à tout moment que la valeur n'est pas égal à 1.

Vous pouvez utiliser ces informations de groupe pour réorganiser ces valeurs avec ave (cela entraîne une transformation intra-groupe).

revsort<-function(...) sort(...,decreasing=TRUE) 
ave(x, grp, FUN=revsort) 
# [1] 3 2 1 7 2 9 8 
5

On pourrait aussi faire:

x <- c(0, which(diff(vec) != 1), length(vec)) 
unlist(sapply(seq(length(x) - 1), function(i) rev(vec[(x[i]+1):x[i+1]]))) 

#[1] 3 2 1 7 2 9 8 

L'idée est d'abord coupé le vecteur sur la base des positions des numéros consécutifs. Nous traversons ensuite ces coupes et appliquons rev.


données

vec <- c(1, 2, 3, 7, 2, 8, 9) 

Analyse comparative

library(microbenchmark) 
vec1 <- c(1, 2, 3, 7, 2, 8, 9) 
vec2 <- c(1:1000, sample.int(100, 10), 5:500, sample.int(100, 10), 100:125) 

f_MrFlick <- function(x){ 
    revsort<-function(...) sort(...,decreasing=TRUE) 
    grp <- cumsum(!c(TRUE, diff(x)==1)) 
    ave(x, grp, FUN=revsort) 
} 

f_989 <- function(vec){ 
    x <- c(0, which(diff(vec) != 1), length(vec)) 
    unlist(sapply(seq(length(x) - 1), function(i) rev(vec[(x[i]+1):x[i+1]]))) 
} 

all(f_MrFlick(vec1)==f_989(vec1)) 
# [1] TRUE 
all(f_MrFlick(vec2)==f_989(vec2)) 
# [1] TRUE 

length(vec1) 
# [1] 7 
microbenchmark(f_MrFlick(vec1), f_989(vec1)) 
# Unit: microseconds 
      # expr  min  lq  mean median  uq  max neval 
# f_MrFlick(vec1) 287.006 297.0585 305.6340 302.833 312.6695 479.912 100 
    # f_989(vec1) 113.348 119.7645 124.7901 121.903 127.0360 268.186 100 
# -------------------------------------------------------------------------- 
length(vec2) 
# [1] 1542 
microbenchmark(f_MrFlick(vec2), f_989(vec2)) 
# Unit: microseconds 
      # expr  min  lq  mean median  uq  max neval 
# f_MrFlick(vec2) 1532.553 1565.2745 1725.7540 1627.937 1698.084 3914.149 100 
    # f_989(vec2) 426.874 454.6765 546.9115 466.439 489.322 2634.383 100