Чередовать списки в R

Допустим, у меня есть два списка в R, не обязательно одинаковой длины, например:

 a <- list('a.1','a.2', 'a.3')
 b <- list('b.1','b.2', 'b.3', 'b.4')

Каков наилучший способ создать список чередующихся элементов, когда после добавления элемента более короткого списка в конце добавляются остальные элементы более длинного списка?

interleaved <- list('a.1','b.1','a.2', 'b.2', 'a.3', 'b.3','b.4')

без использования цикла. Я знаю, что mapply работает для случая, когда оба списка имеют одинаковую длину.

4 ответа

Решение

Вот один из способов:

idx <- order(c(seq_along(a), seq_along(b)))
unlist(c(a,b))[idx]

# [1] "a.1" "b.1" "a.2" "b.2" "a.3" "b.3" "b.4"

Как указывает @James, так как вам нужен список обратно, вы должны сделать:

(c(a,b))[idx]

G. Grothendieck подобный вопрос, я наткнулся на это прекрасное решение Габора Гротендика (то есть G. Grothendieck?) Для некоторых случаев:

c(rbind(a,b))

Это работает одинаково хорошо, когда a а также b оба списка, или когда a а также b оба вектора. Это не точное решение вопроса ОП, потому что когда a а также b иметь разную длину, он будет перерабатывать элементы более короткой последовательности, выводя предупреждение. Однако, поскольку это решение простое и элегантное и дает ответ на очень похожий вопрос - вопрос некоторых людей (таких как я), которые в результате нашли свой путь на эту страницу - его, кажется, стоит добавить в качестве ответа.

Вот один из вариантов, используя interleave функция из ggplot2. Я уверен, что это можно улучшить, но это начало:

require(ggplot2)
Interleave <- function(x,y){
    v <- list(x,y)
    lengths <- sapply(v,length)
    mn <- min(lengths)
    v <- v[order(lengths)]
    c(ggplot2:::interleave(v[[1]],v[[2]][seq_len(mn)]),v[[2]][(mn+1):length(v[[2]])])
}

Interleave(a,b)
Interleave(b,a)

В частности, это будет делать странные вещи, если списки на самом деле имеют одинаковую длину. Может быть, кто-то включит лучший способ сделать индексацию для v[[2]] в последней строке, которая избегает этого вырожденного случая.

interleave(a, b)

# unlist(interleave(a, b))
# [1] "a.1" "b.1" "a.2" "b.2" "a.3" "b.3" "b.4"


interleave <- function(a, b) { 

  shorter <- if (length(a) < length(b)) a else b
  longer  <- if (length(a) >= length(b)) a else b

  slen <- length(shorter)
  llen <- length(longer)


  index.short <- (1:slen) + llen
  names(index.short) <- (1:slen)

  lindex <- (1:llen) + slen
  names(lindex) <- 1:llen


  sindex <- 1:slen
  names(sindex) <- 1:slen

  index <- c(sindex, lindex)
  index <- index[order(names(index))]

  return(c(a, b)[index])

}
a <- list('a.1','a.2', 'a.3')
b <- list('b.1','b.2', 'b.3', 'b.4')

interleave <- function(a, b) {
  mlab <- min(length(a), length(b)) 
  seqmlab <- seq_len(mlab) 
  c(rbind(a[seqmlab], b[seqmlab]), a[-seqmlab], b[-seqmlab]) 
}

interleave(a, b) 

От http://r.789695.n4.nabble.com/Interleaving-elements-of-two-vectors-tp795123p1691409.html

Это немного быстрее, чем Arun:

Другие вопросы по тегам