繁体   English   中英

从R中的2个或更多向量创建嵌套列表?

[英]create nested list from 2 or more vectors in R?

假设我有2个(或更多)向量:

v1 <- letters[1:2]
v2 <- letters[24:26]

> v1
[1] "a" "b"

> v2
[1] "x" "y" "z"

我如何以编程方式构造一个嵌套列表,其中包含输入向量的所有组合的列表? 编辑:我想做到这一点而没有低效的循环结构。

> l1 <- list(list(v1[1], v2[1]), list(v1[1], v2[2]), 
        list(v1[1], v2[3]), list(v1[2], v2[1]), 
        list(v1[2], v2[2]), list(v1[2], v2[3]))

> str(l1)
List of 6
$ :List of 2
..$ : chr "a"
..$ : chr "x"
$ :List of 2
..$ : chr "a"
..$ : chr "y"
$ :List of 2
..$ : chr "a"
..$ : chr "z"
$ :List of 2
..$ : chr "b"
..$ : chr "x"
$ :List of 2
..$ : chr "b"
..$ : chr "y"
$ :List of 2
..$ : chr "b"
..$ : chr "z"

一种简单的方法是使用循环:

l1 <- list()
list_pos <- 1L

for (i in 1:length(v1)) {
  for (j in 1:length(v2)) {
    l1[[list_pos]] <- list(v1[i], v2[j])
    list_pos <- list_pos + 1L
  }  
}

编辑

我认为循环会得到一些不公平的代表,但是下面的一些(丑陋的)代码避免了它们:

l2 <- apply(expand.grid(v1, v2), 1, as.list)

顺便说一句。 循环更快。 我怀疑仍然有一些更漂亮,更富创意的方式来做到这一点。

这是一种将所需组合生成一个数据帧的解决方案。

> v1 <- letters[1:2]
> v2 <- letters[24:26]
> 
> aResult <- expand.grid(v2,v1)
> View(aResult)
> aResult
  Var1 Var2
1    x    a
2    y    a
3    z    a
4    x    b
5    y    b
6    z    b
> 

如果必须将数据帧拆分为每个列表条目一个组合的list() ,我们可以使用split()完成此操作。

#If the data frame must be split into a list, one entry per row
aResult$id <- as.factor(1:nrow(aResult))
theList <- split(aResult,aResult$id)

打印列表中的第一个元素,我们将看到以下内容。

> theList[[1]]
  Var1 Var2 id
1    x    a  1
> 

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM