1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
#' Extract the complete match
#'
#' `str_extract()` extracts the first complete match from each string,
#' `str_extract_all()`extracts all matches from each string.
#'
#' @inheritParams str_detect
#' @param group If supplied, instead of returning the complete match, will
#' return the matched text from the specified capturing group.
#' @seealso [str_match()] to extract matched groups;
#' [stringi::stri_extract()] for the underlying implementation.
#' @param simplify A boolean.
#' * `FALSE` (the default): returns a list of character vectors.
#' * `TRUE`: returns a character matrix.
#' @return
#' * `str_extract()`: an character vector the same length as `string`/`pattern`.
#' * `str_extract_all()`: a list of character vectors the same length as
#' `string`/`pattern`.
#' @export
#' @examples
#' shopping_list <- c("apples x4", "bag of flour", "bag of sugar", "milk x2")
#' str_extract(shopping_list, "\\d")
#' str_extract(shopping_list, "[a-z]+")
#' str_extract(shopping_list, "[a-z]{1,4}")
#' str_extract(shopping_list, "\\b[a-z]{1,4}\\b")
#'
#' str_extract(shopping_list, "([a-z]+) of ([a-z]+)")
#' str_extract(shopping_list, "([a-z]+) of ([a-z]+)", group = 1)
#' str_extract(shopping_list, "([a-z]+) of ([a-z]+)", group = 2)
#'
#' # Extract all matches
#' str_extract_all(shopping_list, "[a-z]+")
#' str_extract_all(shopping_list, "\\b[a-z]+\\b")
#' str_extract_all(shopping_list, "\\d")
#'
#' # Simplify results into character matrix
#' str_extract_all(shopping_list, "\\b[a-z]+\\b", simplify = TRUE)
#' str_extract_all(shopping_list, "\\d", simplify = TRUE)
#'
#' # Extract all words
#' str_extract_all("This is, suprisingly, a sentence.", boundary("word"))
str_extract <- function(string, pattern, group = NULL) {
if (!is.null(group)) {
return(str_match(string, pattern)[, group + 1])
}
check_lengths(string, pattern)
switch(type(pattern),
empty = stri_extract_first_boundaries(string, opts_brkiter = opts(pattern)),
bound = stri_extract_first_boundaries(string, opts_brkiter = opts(pattern)),
fixed = stri_extract_first_fixed(string, pattern, opts_fixed = opts(pattern)),
coll = stri_extract_first_coll(string, pattern, opts_collator = opts(pattern)),
regex = stri_extract_first_regex(string, pattern, opts_regex = opts(pattern))
)
}
#' @rdname str_extract
#' @export
str_extract_all <- function(string, pattern, simplify = FALSE) {
check_lengths(string, pattern)
check_bool(simplify)
switch(type(pattern),
empty = stri_extract_all_boundaries(string,
simplify = simplify, omit_no_match = TRUE, opts_brkiter = opts(pattern)),
bound = stri_extract_all_boundaries(string,
simplify = simplify, omit_no_match = TRUE, opts_brkiter = opts(pattern)),
fixed = stri_extract_all_fixed(string, pattern,
simplify = simplify, omit_no_match = TRUE, opts_fixed = opts(pattern)),
coll = stri_extract_all_coll(string, pattern,
simplify = simplify, omit_no_match = TRUE, opts_collator = opts(pattern)),
regex = stri_extract_all_regex(string, pattern,
simplify = simplify, omit_no_match = TRUE, opts_regex = opts(pattern))
)
}
|
