我有一个带有2列的DF,我有一个单词列表。
list_of_words <- c("tiger","elephant","rabbit", "hen", "dog", "Lion", "camel", "horse")
df <- tibble::tibble(page=c(12,6,9,18,2,15,81,65),
text=c("I have two pets: a dog and a hen",
"lion and Tiger are dangerous animals",
"I have tried to ride a horse",
"Why elephants are so big in size",
"dogs are very loyal pets",
"I saw a tiger in the zoo",
"the lion was eating a buffalo",
"parrot and crow are very clever birds"))
animals <- c("dog,hen", "lion,tiger", "horse", FALSE, "dog", "tiger", "lion", FALSE)
cbind(df, animals)
#> page text animals
#> 1 12 I have two pets: a dog and a hen dog,hen
#> 2 6 lion and Tiger are dangerous animals lion,tiger
#> 3 9 I have tried to ride a horse horse
#> 4 18 Why elephants are so big in size FALSE
#> 5 2 dogs are very loyal pets dog
#> 6 15 I saw a tiger in the zoo tiger
#> 7 81 the lion was eating a buffalo lion
#> 8 65 parrot and crow are very clever birds FALSE
我需要找出列表中的任何一个单词是否存在于DF的一列中。如果是,则将单词/单词返回到DF中的新列。这是单词列表 - &gt;(虎,大象,兔子,母鸡,狗,狮子,骆驼,马)。 This is how my DF Looks like I want something like this
答案 0 :(得分:0)
希望这有帮助!
library(dplyr)
df %>%
rowwise() %>%
mutate(animals = paste(list_of_words[unlist(
lapply(list_of_words, function(x) grepl(x, text, ignore.case = T)))], collapse=",")) %>%
data.frame()
输出为:
page text animals
1 12 pets: dog & hen hen,dog
2 6 Lions and tigers are dangerous animal tiger,Lion
3 9 I have tried to ride a horse horse
4 65 parrot & crow are very clever birds
示例数据:
df <- structure(list(page = c(12, 6, 9, 65), text = structure(c(4L,
2L, 1L, 3L), .Label = c("I have tried to ride a horse", "Lions and tigers are dangerous animal",
"parrot & crow are very clever birds", "pets: dog & hen"), class = "factor")), .Names = c("page",
"text"), row.names = c(NA, -4L), class = "data.frame")
list_of_words <- c("tiger", "elephant", "rabbit", "hen", "dog", "Lion", "camel", "horse")
另一种方法:
library(data.table)
setDT(df)[, animals := paste(list_of_words[unlist(lapply(list_of_words, function(x) grepl(x, text, ignore.case = T)))], collapse = ","), by = 1:nrow(df)]
#> df
# page text animals
#1: 12 pets: dog & hen hen,dog
#2: 6 Lions and tigers are dangerous animal tiger,Lion
#3: 9 I have tried to ride a horse horse
#4: 65 parrot & crow are very clever birds