将不等长列表的列表转换为数据帧

时间:2018-04-21 03:51:20

标签: r json purrr

我有一个从JSON获得的不等长度列表的列表,我想把它变成一个数据帧或一系列数据帧。举个例子,假设这是在原始JSON上使用fromJSON的结果:

mylist <- list(
  list(
    volume = array(1:6, dim=c(3,2)),
    price = array(c(1,2,3,7,8,9), dim=c(3,2)),
    name = 'A'
  ),
  list(
    volume = array(1:10, dim=c(5,2)),
    price = array(c(1:5,12:16), dim=c(5,2)),
    name = 'B'
  ),
  list(
    volume = array(1:14, dim=c(7,2)),
    price = array(c(1:7,21:27), dim=c(7,2)),
    name = 'C'
  )
)

价格和数量列表的长度不等,我想处理数据,假设给定变量的n个观察值是系列中的最后n个条目。因此,对于每个数据列表,我基本上想要丢弃第一列并将数据底部对齐。因此,表示价格数据的一种方法如下:

tribble(
  ~Day, ~PriceA, ~PriceB, ~PriceC,
  #---|--------|--------|---------
    1L,      NA,      NA,      21,
    2L,      NA,      NA,      22,
    3L,      NA,      12,      23,
    4L,      NA,      13,      24,
    5L,       7,      14,      25,
    6L,       8,      15,      26,
    7L,       9,      16,      27
)

如果以这种方式完成,我需要为卷创建一个单独的表。我对其他表示最终数据集的方式持开放态度,例如使用数据框中的嵌套列。

任何人都知道如何优雅地做到这一点?特别令我困惑的是使用purrr / map来操作所涉及的第二级列表。

1 个答案:

答案 0 :(得分:2)

以下是使用tidyverse

的选项
library(tidyverse)
out <- mylist %>%
         transpose %>% 
         map(~ if(all(lengths(.x) == 1)) unlist(.x) else
         map(.x, as_tibble) %>%
           reduce(full_join, by = 'V1') %>%
        mutate_all(funs(.[order(!is.na(.))])))

现在,我们可以提取list元素

out$price %>%
      set_names(c("Day", paste0("Price", LETTERS[1:3])))
# A tibble: 7 x 4
#    Day PriceA PriceB PriceC
#  <dbl>  <dbl>  <int>  <int>
#1  1.00  NA        NA     21
#2  2.00  NA        NA     22
#3  3.00  NA        12     23
#4  4.00  NA        13     24
#5  5.00   7.00     14     25
#6  6.00   8.00     15     26
#7  7.00   9.00     16     27