Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Collapse runs of consecutive numbers to ranges

Tags:

r

Consider the following comma-separated string of numbers:

s <- "1,2,3,4,8,9,14,15,16,19"
s
# [1] "1,2,3,4,8,9,14,15,16,19"

Is it possible to collapse runs of consecutive numbers to its corresponding ranges, e.g. the run 1,2,3,4 above would be collapsed to the range 1-4. The desired result looks like the following string:

s
# [1] "1-4,8,9,14-16,19"
like image 334
user969113 Avatar asked Jun 04 '13 07:06

user969113


2 Answers

I took some heavy inspiration from the answers in this question.

findIntRuns <- function(run){
  rundiff <- c(1, diff(run))
  difflist <- split(run, cumsum(rundiff!=1))
  unlist(lapply(difflist, function(x){
    if(length(x) %in% 1:2) as.character(x) else paste0(x[1], "-", x[length(x)])
  }), use.names=FALSE)
}

s <- "1,2,3,4,8,9,14,15,16,19"
s2 <- as.numeric(unlist(strsplit(s, ",")))

paste0(findIntRuns(s2), collapse=",")
[1] "1-4,8,9,14-16,19"

EDIT: Multiple solutions: benchmarking time!

Unit: microseconds
   expr     min      lq   median       uq      max neval
 spee() 277.708 295.517 301.5540 311.5150 1612.207  1000
  seb() 294.611 313.025 321.1750 332.6450 1709.103  1000
 marc() 672.835 707.549 722.0375 744.5255 2154.942  1000

@speendo's solution is the fastest at the moment, but none of these have been optimised yet.

like image 172
sebastian-c Avatar answered Sep 30 '22 08:09

sebastian-c


I was too slow... but here's another solution.

It uses less R-specific functions so it could be ported to other languages (on the other hand maybe it's less elegant)

s <- "1,2,3,4,8,9,14,15,16,19"

collapseConsecutive <- function(s){
  x <- as.numeric(unlist(strsplit(s, ",")))

  x_0 <- x[1]
  out <- toString(x[1])
  hasDash <- FALSE

  for(i in 2:length(x)) {
    x_1 <- x[i]
    x_2 <- x[i+1]

    if((x_0 + 1) == x_1 && !is.na(x_2) && (x_1 + 1) == x_2) {
      if(!hasDash) {
        out <- c(out, "-")
        hasDash <- TRUE
      }
    } else {
      if(hasDash) {
        hasDash <- FALSE
      } else {
        out <- c(out, ",")
      }
      out <- c(out, x_1)
      hasDash <- FALSE
    }
    x_0 <- x_1
  }
  outString <- paste(out, collapse="")
  outString
}

collapseConsecutive(s)
# [1] "1-4,8,9,14-16,19"
like image 38
speendo Avatar answered Sep 30 '22 07:09

speendo