使用 dplyr mutate 将向量分成两列

Question

假设我有以下功能

get_answer <- function(condition, dp, rp){
  if(condition == "DD"){
    result <- rbinom(n = 2, size = 1, prob = dp)
  }
  
  if(condition %in% c("DR", "RD")){
    result <- c(rbinom(n = 1, size = 1, prob = dp), 
                rbinom(n = 1, size = 1, prob = rp))
  }
  
  if(condition == "RR"){
    result <- rbinom(n = 2, size = 1, prob = rp)
  }
  
  return(result)
}

我像这样创建一个data.frame：

results_df <- data.frame(condition = c(rep("DD", 10000), rep("DR", 10000), rep("RR", 10000)))

我希望能够获取从get_answer 返回的向量，用于condition 列中的条件，并将返回值分成两列——第一个值进入列P1，第二个值进入列P2。

类似这样的：

results_df %>% mutate(p1 = get_answer(condition, .6, .4)[0], p2 = get_answer(condition, .6, .4)[1])

在 dplyr 中执行此操作的正确方法是什么？

Answer 1

该函数未矢量化。因此，我们需要使用rowwise 在每一行上应用它。此外，R 中的索引从 1 开始，而不是 0

library(dplyr)
results_df %>%
     rowwise %>%
     mutate(p1 = get_answer(condition, .6, .4)[1], 
             p2 = get_answer(condition, .6, .4)[2])

---

我们可以有一个list 列，然后使用来自tidyr 的unnest_wider，而不是调用该函数两次

library(tidyr)
library(stringr)
out <- results_df %>%
    rowwise %>%
    mutate(p1 = list(get_answer(condition, .6, .4) %>%
           as.list)) %>%
    ungroup %>%
    unnest_wider(c(p1)) %>%
    rename_at(-1, ~ str_c('p', seq_along(.)))