如何遍历一列的字符串以变异/修改其他列的值？

Question

我有一个数据帧，其中一列（x1）包含字符串值。我使用这些字符串值修改数据帧中其他列的相应逻辑值（其他列以列x1的字符串中可能的子字符串值命名，即“Dog”、“Cat”、“Bird”）。

我已经弄清楚了如何使用str裂（）来解析x1列中的每个字符串。我还知道如何根据这些str裂（）结果改变/修改其他三列。

我目前坚持的是如何将下面的for循环应用于我的数据帧中的每一行。

x0 <- c(1,2,3,4,5)
x1 <- c("Dog, Cat", "Cat", "Dog, Bird", "Cat, Bird, Dog", "Cat, Bird")
Dog <- c(rep(FALSE, 5))
Cat <- c(rep(FALSE, 5))
Bird <- c(rep(FALSE, 5))

example_df <- data.frame(x0, x1, Dog, Cat, Bird)

for(i in 1:length(strsplit(example_df$x1) )){
  example_df[[strsplit(example_df$x1)[i]]] <- TRUE
}

因此，对于上面的示例，我希望我的代码将数据帧的第一行更改为example_df$Dog和example_df$Cat都为TRUE，但example_df$Bird对于该行仍然为FALSE。第二行只会example_df$Cat为TRUE，等等。

另请注意：对于上面的示例，我只提供了三个动物字符串值。但我正在寻找一种方法，它可以对大量字符串值进行适当的扩展。我知道可以通过这种方法使用复制和粘贴：

example_df %>%
    mutate(Dog = str_contains(x1, "Dog"))

但不幸的是，如果我在x1列中有10或20个可能的动物子字符串值，则此方法不会很好地扩展。

Answer 1

我们可以在str拆分之后使用mtable

library(qdapTools)
m1 <- mtabulate(strsplit(example_df$x1, ",\\s+")) > 0
example_df[names(example_df)[3:5]] <- m1[, names(example_df)[3:5]]

-输出

example_df
#  x0             x1   Dog   Cat  Bird
#1  1       Dog, Cat  TRUE  TRUE FALSE
#2  2            Cat FALSE  TRUE FALSE
#3  3      Dog, Bird  TRUE FALSE  TRUE
#4  4 Cat, Bird, Dog  TRUE  TRUE  TRUE
#5  5      Cat, Bird FALSE  TRUE  TRUE

或者另一个选项是

library(splitstackshape)
cSplit_e(example_df[1:2], "x1", sep = ",", type = "character", fill = 0, drop = TRUE)

Answer 2

根据你的例子调整这个答案。你可以使用separate_rows然后pivot_wider。

library(tidyverse)

example_df %>%
  select(x0, x1) %>%
  separate_rows(x1) %>%
  pivot_wider(names_from = x1, 
              values_from = x1, 
              values_fn = list(x1 = is.character),
              values_fill = list(x1 = FALSE))


#----------
# A tibble: 5 x 4
     x0 Dog   Cat   Bird 
  <dbl> <lgl> <lgl> <lgl>
1     1 TRUE  TRUE  FALSE
2     2 FALSE TRUE  FALSE
3     3 TRUE  FALSE TRUE 
4     4 TRUE  TRUE  TRUE 
5     5 FALSE TRUE  TRUE