从R中的字符串中提取单词

时间:2016-11-26 23:33:17

标签: r regex pattern-matching stringr

我正在尝试提取字符串的片段并从这些匹配的模式中创建新变量。我已经尝试过" strings"中的许多函数。包装似乎无法得到结果。以下示例由数据组成。我想获取一个字符串并提取它们并将它们存储到新数据框的新列中。

例如

ex <- c("The Accountant (2016)Crime (vodmovies112.blogspot.com.es)","Miss Peregrine's Home for Peculiar Children (2016)FantasySci-Fi (vodmovies112.blogspot.com.es),"Fantastic Beasts And Where To Find Them (2016) TSAdventure (openload.co)","Ben-Hur (2016) HDActionAdventure (vodmovies112.blogspot.com.es)","The Remains (2016) 1080p BlurayHorror (openload.co)" ,"Suicide Squad (2016) HDAction (openload.co)")

     >ex
[1] "The Accountant (2016)Crime (vodmovies112.blogspot.com.es)"
[2] "Miss Peregrine's Home for Peculiar Children (2016)FantasySci-Fi (vodmovies112.blogspot.com.es)"
[3] "Fantastic Beasts And Where To Find Them (2016) TSAdventure (openload.co)"
[4] "Ben-Hur (2016) HDActionAdventure (vodmovies112.blogspot.com.es)"
[5] "The Remains (2016) 1080p BlurayHorror (openload.co)"
[6] "Suicide Squad (2016) HDAction (openload.co)"

genres <- c("Action","Adventure","Animation","Biography",
        "Comedy","Crime","Documentary","Drama","Family",
        "Fantasy","Film-Noir","History","Horror","Music",
        "Musical","Mystery","Romance","Sci-Fi","Sport","Thriller",
        "War","Western")

genres <- paste0("^",genres,"|")
genres[22] <- "^Western"
> genres
[1] "^Action|"      "^Adventure|"   "^Animation|"   "^Biography|"
[5] "^Comedy|"      "^Crime|"       "^Documentary|" "^Drama|"
[9] "^Family|"      "^Fantasy|"     "^Film-Noir|"   "^History|"
[13] "^Horror|"      "^Music|"       "^Musical|"     "^Mystery|"
[17] "^Romance|"     "^Sci-Fi|"      "^Sport|"       "^Thriller|"
[21] "^War|"         "^Western"

试图完成

> df
           title year                       domain genre
1 The Accountant 2016 vodmovies112.blogspot.com.es Crime

2 个答案:

答案 0 :(得分:0)

这是一种可能性:

temp <- strsplit(ex, "\\(|\\)")
df <- setNames(as.data.frame(lapply(1:4,function(i) sapply(temp,"[",i)), stringsAsFactors = FALSE), c("title", "year", "genre", "domain"))
df <- df[ , c("title", "year", "domain", "genre")]
correct <- sapply(seq_along(df$genre), function(y) which(lengths(sapply(seq_along(genres), function(x) grep(genres[x], df$genre[y])))>0))
correct <- lapply(correct, function(x) paste0(genres[x], collapse = " "))
df$genre <- unlist(correct)

df
#                                         title year                       domain            genre
# 1                              The Accountant  2016 vodmovies112.blogspot.com.es            Crime
# 2 Miss Peregrine's Home for Peculiar Children  2016 vodmovies112.blogspot.com.es   Fantasy Sci-Fi
# 3     Fantastic Beasts And Where To Find Them  2016                  openload.co        Adventure
# 4                                     Ben-Hur  2016 vodmovies112.blogspot.com.es Action Adventure
# 5                                 The Remains  2016                  openload.co           Horror
# 6                               Suicide Squad  2016                  openload.co           Action

基本上,我们将矢量ex分成4个部分,用括号分隔。然后,我们使用4列创建data.frame df。 最难的部分是正确提取类型(因为每部电影可能有不止一种类型)。我使用sapplylapplygrep的组合来执行此操作。完成后,我们“纠正”列类型。

以下是您的数据:

ex <- c("The Accountant (2016)Crime (vodmovies112.blogspot.com.es)", 
"Miss Peregrine's Home for Peculiar Children (2016)FantasySci-Fi (vodmovies112.blogspot.com.es)", 
"Fantastic Beasts And Where To Find Them (2016) TSAdventure (openload.co)", 
"Ben-Hur (2016) HDActionAdventure (vodmovies112.blogspot.com.es)", 
"The Remains (2016) 1080p BlurayHorror (openload.co)", "Suicide Squad (2016) HDAction (openload.co)"
)

genres <- c("Action", "Adventure", "Animation", "Biography", "Comedy", 
"Crime", "Documentary", "Drama", "Family", "Fantasy", "Film-Noir", 
"History", "Horror", "Music", "Musical", "Mystery", "Romance", 
"Sci-Fi", "Sport", "Thriller", "War", "Western")

答案 1 :(得分:0)

使用 tidyverse 的另一种可能性:

library(tidyverse)

data_frame(x = ex) %>%
    extract(
        x,
        c("title", "year", "domain", "genre"), 
        "(^[^(]+)\\s+\\((\\d{4})\\)\\s*([^(]+)\\s+\\(([^)]+)"
    )

##                                         title  year             domain                        genre
## *                                       <chr> <chr>              <chr>                        <chr>
## 1                              The Accountant  2016              Crime vodmovies112.blogspot.com.es
## 2 Miss Peregrine's Home for Peculiar Children  2016      FantasySci-Fi vodmovies112.blogspot.com.es
## 3     Fantastic Beasts And Where To Find Them  2016        TSAdventure                  openload.co
## 4                                     Ben-Hur  2016  HDActionAdventure vodmovies112.blogspot.com.es
## 5                                 The Remains  2016 1080p BlurayHorror                  openload.co
## 6                               Suicide Squad  2016           HDAction                  openload.co