我有一个列表，其中的元素要么包含单个整数，要么包含整数向量.
int_list <- list(
c(1, 15),
c(3, 19),
c(2, 16),
c(4, 19),
c(5, 21),
c(19, 28),
c(28, 30),
17
)
如您所见，第二个和第四个元素包含整数19.我想将这些情况合并到一个向量中.因此，在这种情况下预期/期望的结果将是:
int_list_merged <- list(
c(1, 15),
c(3, 4, 19, 28, 30),
c(2, 16),
c(5, 21),
17
)
我怎样才能做到这一点呢？
list(c(1,2), c(2,3), c(3,4))返回list(c(1,2,3,4))的条件意味着我们必须遍历列表，直到它停止更改.
我们可以使用递归函数，该函数在满足该条件时停止调用自身.
create_merged_list <- function(l, check_finished_list = NULL) {
new_l <- unique(lapply(seq(l), \(i) merge_elements(l, i)))
if (identical(check_finished_list, new_l)) {
return(new_l)
}
create_merged_list(new_l, l)
}
这次调用的主要函数是merge_elements()，它迭代列表的每个元素，将其与与其共享值的其他元素合并，并返回一个新的列表.
merge_elements <- function(l, i) {
l_compare <- l[-i]
el <- l[[i]]
match_vals <- which(outer(el, unlist(l_compare), \(x, y) x == y), arr.ind = TRUE)[, "col"]
if (!length(match_vals)) {
return(el)
}
l_breaks <- cumsum(lengths(l_compare))
l_match_idx <- vapply(match_vals, \(x) min(which(x <= l_breaks)), integer(1))
new_el <- sort(unique(c(el, unlist(l_compare[l_match_idx]))))
new_el
}
create_merged_list(int_list)
# [[1]]
# [1] 1 15
# [[2]]
# [1] 3 4 19 28 30
# [[3]]
# [1] 2 16
# [[4]]
# [1] 5 21
# [[5]]
# [1] 17
identical(create_merged_list(int_list), int_list_merged)
# [1] TRUE