R - 提取匹配模式的所有字符串并创建关系表

时间:2016-07-28 09:07:20

标签: regex r dplyr

我正在寻找一个更短,更漂亮的解决方案(可能在tidyverse)以解决以下问题。我有一个data.frame“数据”:

  id            string
1  A 1.001 xxx 123.123
2  B 23,45 lorem ipsum
3  C      donald trump
4  D    ssss 134, 1,45

我想要做的是提取所有数字(无论分隔符是“。”还是“,” - >在这种情况下我假设字符串“134,1,45”可以提取为两个数字:134和1.45)并创建一个与此类似的data.frame“output”:

  id  string
1  A   1.001
2  A 123.123
3  B   23.45
4  C    <NA>
5  D     134
6  D    1.45

我设法做到了这一点(下面的代码),但解决方案对我来说非常难看,也没那么高效(两个for循环)。有人可以建议一个更好的方法(最好使用dplyr)

# data
data <- data.frame(id = c("A", "B", "C", "D"), 
                  string = c("1.001 xxx 123.123", 
                             "23,45 lorem ipsum", 
                             "donald trump", 
                             "ssss 134, 1,45"),
                  stringsAsFactors = FALSE)

# creating empty data.frame                     
len <- length(unlist(sapply(data$string, function(x) gregexpr("[0-9]+[,|.]?[0-9]*", x))))
output <- data.frame(id = rep(NA, len), string = rep(NA, len))

# main solution
start = 0

for(i in 1:dim(data)[1]){
  tmp_len <- length(unlist(gregexpr("[0-9]+[,|.]?[0-9]*", data$string[i])))
  for(j in (start+1):(start+tmp_len)){
    output[j,1] <- data$id[i]
    output[j,2] <- regmatches(data$string[i], gregexpr("[0-9]+[,|.]?[0-9]*", data$string[i]))[[1]][j-start]
  }
  start = start + tmp_len
}

# further modifications
output$string <- gsub(",", ".", output$string)
output$string <- as.numeric(ifelse(substring(output$string, nchar(output$string), nchar(output$string)) == ".",
                                   substring(output$string, 1, nchar(output$string) - 1),
                                   output$string))

output

3 个答案:

答案 0 :(得分:5)

1)Base R 这使用相对简单的正则表达式而没有包。

在前两行代码中,用逗号替换任何后跟空格的逗号 空格然后用点替换所有剩余的逗号。这两行s之后将是:c("1.001 xxx 123.123", "23.45 lorem ipsum", "donald trump", "ssss 134 1.45")

在接下来的4行代码中,从每个字符串字段的开头和结尾修剪空格,并在空格上分割字符串字段, 名单。 grep表示仅由数字和点组成的元素。 (正则表达式^[0-9.]*$匹配单词的开头,后跟零个或多个数字或点,后跟单词的结尾,因此只匹配包含这些字符的单词。)用NA替换任何零长度的组件。最后添加data$id作为名称。运行这4行后,列表L将为list(A = c("1.001", "123.123"), B = "23.45", C = NA, D = c("134", "1.45"))

在最后一行代码中,将列表L转换为具有适当名称的数据框。

s <- gsub(", ", " ", data$string)
s <- gsub(",", ".", s)

L <- strsplit(trimws(s), "\\s+")
L <- lapply(L, grep, pattern = "^[0-9.]*$", value = TRUE)
L <- ifelse(lengths(L), L, NA)
names(L) <- data$id

with(stack(L), data.frame(id = ind, string = values))

,并提供:

  id  string
1  A   1.001
2  A 123.123
3  B   23.45
4  C    <NA>
5  D     134
6  D    1.45

2)magrittr (1)的这种变体将其写为magrittr管道。

library(magrittr)

data %>%
     transform(string = gsub(", ", " ", string)) %>%
     transform(string = gsub(",", ".", string)) %>%
     transform(string = trimws(string)) %>%
     with(setNames(strsplit(string, "\\s+"), id)) %>%
     lapply(grep, pattern = "^[0-9.]*$", value = TRUE) %>%
     replace(lengths(.) == 0, NA) %>%
     stack() %>%
     with(data.frame(id = ind, string = values))

3)dplyr / tidyr 这是使用dplyr和tidyr的备用管道解决方案。 unnest转换为长格式,id是因素,以便我们以后可以使用complete来恢复后续过滤删除的ID,过滤器会删除垃圾行和complete为每个id插入不会出现的NA行。

library(dplyr)
library(tidyr)

data %>%
  mutate(string = gsub(", ", " ", string)) %>%
  mutate(string = gsub(",", ".", string)) %>%
  mutate(string = trimws(string)) %>%
  mutate(string = strsplit(string, "\\s+")) %>%
  unnest() %>%
  mutate(id = factor(id))
  filter(grepl("^[0-9.]*$", string)) %>%
  complete(id)

4)data.table

library(data.table)

DT <- as.data.table(data)
DT[, string := gsub(", ", " ", string)][, 
     string := gsub(",", ".", string)][,
     string := trimws(string)][,
     string := setNames(strsplit(string, "\\s+"), id)][,
     list(string = list(grep("^[0-9.]*$", unlist(string), value = TRUE))), by = id][,
     list(string = if (length(unlist(string))) unlist(string) else NA_character_), by = id]
DT

更新删除了垃圾字没有数字或点的假设。还添加了(2),(3)和(4)以及一些改进。

答案 1 :(得分:2)

我们可以使用,(使用.)替换数字之间的gsub,使用str_extract_all(从stringr提取数字到{ {1}}),用list替换list等于0的length元素,将NA的名称设置为&#39; id&#39;列,liststack转换为list并重命名列。

data.frame

答案 2 :(得分:1)

与Gabor相同的想法。我原本希望使用R的内置字符串解析(type.convert,用于read.table)而不是编写自定义正则表达式替换:

sp = setNames(strsplit(data$string, " "), data$id)

spc = lapply(sp, function(x) {
  x = x[grep("[^0-9.,]$", x, invert=TRUE)]

  if (!length(x))
    NA_real_
  else 
    mapply(type.convert, x, dec=gsub("[^.,]", "", x), USE.NAMES=FALSE)
})

setNames(rev(stack(spc)), names(data))

  id  string
1  A   1.001
2  A 123.123
3  B   23.45
4  C    <NA>
5  D     134
6  D    1.45

不幸的是,type.convert不足以同时考虑两个小数分隔符,所以我们需要这个mapply malarkey而不是type.convert(x, dec = "[.,]")