r - 如何将列表元素提取到 r 中的多个 tibble 列中？

Question

我有一个非常大的 tibble 形式的数据集。我想使用一些返回列表的函数来总结数据。我对列表的几个组件感兴趣，我想将我需要的每个组件返回到新的 tibble 列中。

举个例子

library(tibble)
library(dplyr)

# Create a data set of 1,000 random values in 100 subgroups with sample size 10
contrived_data <- tibble(subgroup = rep(1:100, each = 10),
                         value    = rnorm(1000, mean = 5, sd = 1))


# Run the KS test vs. normal distribution on each sample of size 10. Return the KS statistic and p-value
# into new tibble columns
contrived_data %>% group_by(subgroup) %>%
  summarize(avg     = mean(value),
            std_dev = sd(value),
            ks_stat = ks.test(value, "pnorm", mean = 5, sd = 1)$statistic,
            ks_pval = ks.test(value, "pnorm", mean = 5, sd = 1)$p.value)

以这种方式运行它可以得到我想要的结果，但效率不高。调用 ks.test 函数两次意味着执行时间(几乎)加倍。似乎必须有一种更有效的方法来通过单个函数调用提取这两个列表组件，但我不知道该怎么做。

Answer 1

您可以定义函数并使用来自 purrr 的 map :

library(tibble)
library(dplyr)
library(purrr)

func = function(DA){
kstest = ks.test(DA$value, "pnorm", mean = 5, sd = 1)
data.frame(
subgroup = unique(DA$subgroup),
avg=mean(DA$value),
std_dev = sd(DA$value),
ks_stat = kstest$statistic,
ks_pval = kstest$p.value)
}

contrived_data %>% 
split(.$subgroup) %>%
map_dfr(func)