Чередовать списки в 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: