r - 使用 plyr::mutate 以矢量化方式提取列表部分？

Question

假设这段代码:

foo <- data.frame(cols_val=c("NA", "1:2:3", "4:5:6"))
library(plyr)
foo <- mutate(
  foo,
  cols_list = str_split(cols_val, ":"),
  one = cols_list[1],
  two = cols_list[2])

我希望 foo$one 成为 c(NA, "1", "4") 并且 foo$two 成为c(NA, "2", "5")。也就是说，将 cols_val 的值拆分为数据框的各个列。

但是，cols_list 是一个列表，one == cols_list[1] 是该列表的第一个元素(== cols_list[[1] ])，而 两个 == cols_list[1]。所以，我不知道如何正确地对其进行矢量化。

帮忙吗？

> foo
  cols_val cols_list one     two
1       NA        NA  NA 1, 2, 3
2    1:2:3   1, 2, 3  NA 1, 2, 3
3    4:5:6   4, 5, 6  NA 1, 2, 3

> str(foo$cols_list)
List of 3
 $ : chr "NA"
 $ : chr [1:3] "1" "2" "3"
 $ : chr [1:3] "4" "5" "6"

Answer 1

您可以使用 tidyr::separate() 而不是 plyr。它做的正是你想做的。我们可以保留现有列(使用 remove = FALSE)并将所有新列转换为适当的类型(使用 convert = TRUE)。我只使用 fill = "left" 来避免在不使用时出现警告。我不太确定为什么会出现警告。

tidyr::separate(foo, cols_val, c("one", "two", "three"), ":", 
    remove = FALSE, convert = TRUE, fill = "left")
#   cols_val one two three
# 1       NA  NA  NA    NA
# 2    1:2:3   1   2     3
# 3    4:5:6   4   5     6