将列表分成长度为x的列表

时间:2020-06-11 19:47:14

标签: r list split purrr

简单的问题,给出一个列表:

main_list <- list(1:3,
                  4:6,
                  7:9,
                  10:12,
                  13:15)
main_list
# [[1]]
# [1] 1 2 3

# [[2]]
# [1] 4 5 6

# [[3]]
# [1] 7 8 9

# [[4]]
# [1] 10 11 12

# [[5]]
# [1] 13 14 15

我想将列表分成多个列表,然后将原始列表分成长度为x的列表。因此,如果我说x = 2,我会得到3个长度为2、2和剩余1的列表:

target <- list(list(1:3,
               4:6),
               list(7:9,
               10:12),
               list(13:15))
target
# [[1]]
# [[1]][[1]]
# [1] 1 2 3

# [[1]][[2]]
# [1] 4 5 6


# [[2]]
# [[2]][[1]]
# [1] 7 8 9

# [[2]][[2]]
# [1] 10 11 12


# [[3]]
# [[3]][[1]]
# [1] 13 14 15

类似的东西:

my_split <- function(listtest, x) {
  split(listtest, c(1:x))
  }
target <- my_split(main_list, 2)

谢谢

2 个答案:

答案 0 :(得分:4)

这是gl

的选项
split(main_list, as.integer(gl(length(main_list), 2, length(main_list))))

可以将其转换为自定义函数

f1 <- function(lstA, n) {
      l1 < length(lstA)
      split(lstA, as.integer(gl(l1, n, l1)))
  }

答案 1 :(得分:2)

编辑:不需要条件逻辑。只需将split()c()rep()一起使用:

my_split <- function(l, x){

  l_length <- length(l)
  l_div <- l_length / x

  split(l, c(rep(seq_len(l_div), each = x), rep(ceiling(l_div), l_length %% x)))

}

my_split(main_list, 2)