stringi包中stringr :: word的等效函数

时间:2017-11-13 12:07:40

标签: r stringr stringi

我查看了stringi包手册,找到了包word()中函数stringr的等效内容,但我找不到它。我正在寻找它的原因是因为我想为我的语言环境设置排序规则选项,而stringr并未向我提供stringi所做的所有选项。 stringi包中是否有一个与stringr包中的word()函数相同的函数?请参阅下面的示例,其中我想要以“José”一词后面的下三个单词,这些单词以不同的方式编写。在stringi包中,函数stri_opts_collator允许设置参数strength=1以获得允许的校对强度,因此所有形式的“José”都会被考虑在内。

sentencas<-c("josé é um trabalhador responsável","José é um trabalhador 
responsável","jose é um trabalhador responsável","Jose é um trabalhador 
responsável")

sentencas %>% word(2,sep=coll("jose ",ignore_case=T)) %>% word(1,3)

使用stringi,我可以提取所有形式的“José”:

stri_extract_first_coll(sentencas,"jose",strength=1).

我想做这样的事情:

sentencas %>% word(2,sep=coll("jose ",strength=1)) %>% word(1,3)

3 个答案:

答案 0 :(得分:2)

stringr::word()不完全相同,但这似乎可以解决问题:

sentencas %>% 
  stri_split_coll("jose ", strength=1, simplify = TRUE) %>% 
  .[,2] %>% 
  word(1,3)

答案 1 :(得分:2)

这是另一种选择。请注意,我在你的标记词之前引入了一些单词,即“Josè”,因为我假设你不仅会有以这个词开头的句子。重要的功能是使用stri_locate来检测标记词的结尾,添加一个字符位置,并将此位置与字符串中的最后一个字符位置一起传递给stri_sub。请注意,所提供的功能不是故障保护,例如,如果没有匹配等情况

请查看目前为止提出的三种解决方案的以下基准。

library(stringi)
library(magrittr)

sentencas<- c("some words josé é um trabalhador responsável"
              ,"only one word José é um trabalhador responsável"
             ,"several words jose é um trabalhador responsável"
             ,"and again some words Jose é um trabalhador responsável")

stri_word <- function(marker, str, words_after_marker) {

  stri_sub(str,  cbind(stri_locate_first_coll(str, marker ,strength=1)[, "end"]+1, nchar(str)) ) %>% 
  { gsub( "^\\s+|\\s+$", "", stri_extract_first_regex(., paste0("(\\s\\w+){", words_after_marker[1], ",", words_after_marker[2],"}"))) }

}

stri_word("jose", sentencas, c(1,3) )

#[1] "é um trabalhador" "é um trabalhador" "é um trabalhador" "é um trabalhador"

#Benchmarks
library(microbenchmark)
library(stringr)

stringi_positions <- function() {

  stri_word <- function(marker, str, words_after_marker) {

    stri_sub(str,  cbind(stri_locate_first_coll(str, marker ,strength=1)[, "end"]+1, nchar(str)) ) %>%
    { gsub( "^\\s+|\\s+$", "", stri_extract_first_regex(., paste0("(\\s\\w+){", words_after_marker[1], ",", words_after_marker[2],"}"))) }

  }

  stri_word("jose", sentencas, c(1,3) )


}

stringi_map <- function() {

  sentencas %>%
    map(stri_extract_all_words) %>% 
    map(~{
      .x <- flatten_chr(.x)
      map(.x, stri_detect_coll, "jose", ignore.case=TRUE, strength=1L) %>% 
        flatten_lgl() %>%
        which() -> pos
      .x[(pos+1):(pos+1+3)]
    })


}


semi_stringi <- function() {

  sentencas %>% 
    stri_split_coll("jose ", strength=1, simplify = TRUE) %>% 
    .[,2] %>% 
    word(1,3)


}


microbenchmark(
  stringi_map(),
  semi_stringi(),
  stringi_positions()
)

# Unit: microseconds
#                        expr      min       lq      mean    median        uq       max neval
# stringi_map()       3498.667 3752.886 4059.0339 4038.0925 4214.3480  7365.635   100
# semi_stringi()      485.543  558.966  805.0216  593.9015  652.7195 15806.567   100
# stringi_positions()  288.958  325.669  456.9946  344.6180  384.4865 10719.428   100

答案 2 :(得分:1)

word()是一个相当复杂的函数,stringi中没有单个函数等价。这是实现理想结果的另一种方法:

library(stringi)
library(purrr)

sentencas <- c("josé é um trabalhador responsável",
               "José é um trabalhador responsável",
               "jose é um trabalhador responsável",
               "Jose é um trabalhador responsável")

sentencas %>%
  map(stri_extract_all_words) %>% 
  map(~{
    .x <- flatten_chr(.x)
    map(.x, stri_detect_coll, "jose", ignore.case=TRUE, strength=1L) %>% 
      flatten_lgl() %>%
      which() -> pos
    .x[(pos+1):(pos+1+3)]
  })
## [[1]]
## [1] "é"           "um"          "trabalhador" 
## 
## [[2]]
## [1] "é"           "um"          "trabalhador"
## 
## [[3]]
## [1] "é"           "um"          "trabalhador" 
## 
## [[4]]
## [1] "é"           "um"          "trabalhador"