r - 交替组合两个向量

Question

从这篇文章，Alternate, interweave or interlace two vectors ，显示了一种交替组合两个向量的便捷方法:

x <- 1:3 ; y <- 4:6
c(rbind(x, y)) # [1] 1 4 2 5 3 6

但是这种方法只适用于两个向量长度相同的情况。现在我想找到一种通用的方法来做到这一点。

这是我的想法:

fun <- function(x, y, n) {...}

x和 y是两个相同类型的向量，它们的长度可以不同。

n表示交替间隔， .

预期输出:

x <- 1:5 ; y <- c(0, 0, 0)
fun(x, y, n = 1) # 1 0 2 0 3 0 4 5
fun(x, y, n = 2) # 1 2 0 3 4 0 5 0
fun(x, y, n = 5) # 1 2 3 4 5 0 0 0

感谢帮助。

Answer 1

第一次拆分 x成组n并将其堆叠成一个长的data.frame。同样堆栈y然后 rbind将这些数据帧放在一起，按 ind 对它们进行排序(组号)并取values柱子。

fun <- function(x, y, n) {
  sx <- stack(split(x, c(gl(length(x), n, length(x)))))
  sy <- stack(as.list(setNames(y, seq_along(y))))
  r <- rbind(sx, sy)
  r[order(r$ind), "values"]
}

x <- 1:5 ; y <- c(0, 0, 0)

fun(x, y, 1)
## [1] 1 0 2 0 3 0 4 5
fun(x, y, 2)
## [1] 1 2 0 3 4 0 5 0
fun(x, y, 5)
## [1] 1 2 3 4 5 0 0 0