主页 / coding / 问题 / 78915687
Accepted
scjForR
Asked: 2024-08-27 01:34:57 +0800 CST

使用多列从数据框创建嵌套列表

  • 772

我有一个数据框，我想按行将其转换为嵌套列表。但是，我需要将多个列分组到每个嵌套元素中的单独矩阵中。我见过的所有解决方案都将每列放入一个唯一元素中。我想使用列值创建单独的矩阵。

例如，如果我有如下数据框：

df = data.frame(a=c("foo", "bar", "baz"), b=1:3, c=3:1, d=4:6, e=6:4)

我想将其转换为如下形式：

dflist <- list(foo=list(as.matrix(c(1,3)),
                        as.matrix(c(4,6))),
               bar=list(as.matrix(c(2,2)),
                        as.matrix(c(5,5))),
               baz=list(as.matrix(c(3,1)),
                        as.matrix(c(6,4))))

3 个回答

  1. Best Answer

    使用purrr：

    mylist <- pmap(df, \(b, c, d, e, ...) {
  list(as.matrix(c(b, c)), as.matrix(c(d, e)))
}) |> setNames(df$a)
    • 1

  2. 在基础 R 中，您可以split使用以下方法为每个方法创建列表lapply：

    ll <- lapply(split(df, df$a), \(x) list(matrix(x[2:3]), matrix(x[4:5])))

#If order matters (above is alphabetical (bar, bar, foo):
ll_ordered <- ll[match(df$a, names(ll))]

    或者用一行来表达，感谢@Friede 的宝贵建议：

    lapply(split(df, factor(f <- df$a, f)), \(x) list(matrix(x[2:3]), matrix(x[4:5])))

    两种方法均给出输出：

    # $foo
# $foo[[1]]
# [,1]
# [1,] 1   
# [2,] 3   
# 
# $foo[[2]]
# [,1]
# [1,] 4   
# [2,] 6   
# 
# 
# $bar
# $bar[[1]]
# [,1]
# [1,] 2   
# [2,] 2   
# 
# $bar[[2]]
# [,1]
# [1,] 5   
# [2,] 5   
# 
# 
# $baz
# $baz[[1]]
# [,1]
# [1,] 3   
# [2,] 1   
# 
# $baz[[2]]
# [,1]
# [1,] 6   
# [2,] 4
    • 1

  3. 对于每一行，在新的列中创建矩阵列表data，然后提取该列。请注意，在问题中，输入是整数，但dflist包含双精度数。我们假设这不是预期的，而是使用dflist末尾注释中定义的值。

    library(dplyr)

result <- df %>%
  rowwise %>%
  mutate(data = list(cbind(c(b, c)), cbind(c(d, e))) %>%
    list %>%
    setNames(a)) %>%
  pull(data)

identical(result, dflist)
## [1] TRUE

    笔记

    dflist <- list(foo=list(as.matrix(c(1L,3L)),
                        as.matrix(c(4L,6L))),
               bar=list(as.matrix(c(2L,2L)),
                        as.matrix(c(5L,5L))),
               baz=list(as.matrix(c(3L,1L)),
                        as.matrix(c(6L,4L))))
    • 0

相关问题