Search code examples
rlistdataframerow

R - remove rows from data frame that do not match (exactly) elements of list


Imagine a data frame...

df <- rbind("A*YOU 1.000 0.780", "A*YOUR 1.000 0.780", "B*USE 0.800 0.678", "B*USER 0.700 1.000")
df <- as.data.frame(df)
df

... which prints...

> df
                  V1
1  A*YOU 1.000 0.780
2 A*YOUR 1.000 0.780
3  B*USE 0.800 0.678
4 B*USER 0.700 1.000

... and of which I would like to remove any row that does not contain exactly any element of a list (called tenables here) tenables <- c("A*YOU", "B*USE"), so that the outcome becomes:

> df
                  V1
1  A*YOU 1.000 0.780
2  B*USE 0.800 0.678

Any ideas on how to solve this? Many thanks in advance.


Solution

  • Since you have regex specials in tenables (* means "0 or more of the previous character/class/group"), we cannot use fixed=TRUE in the grep call. As such, we need to find those specials and backslash-escape them. From there, we'll add \\b (word-boundary) to differentiate between YOU and YOUR, where adding a space or any other character may be over-constraining.

    ## clean up tenables to be regex-friendly and precise
    gsub("([].*+(){}[])", "\\\\\\1", tenables)
    # [1] "A\\*YOU" "B\\*USE"
    
    ## combine into a single pattern for simple use in grep
    paste0("\\b(", paste(gsub("([].*+(){}[])", "\\\\\\1", tenables), collapse = "|"), ")\\b")
    # [1] "\\b(A\\*YOU|B\\*USE)\\b"
    
    ## subset your frame
    subset(df, !grepl(paste0("\\b(", paste(gsub("([].*+(){}[])", "\\\\\\1", tenables), collapse = "|"), ")\\b"), V1))
    #                   V1
    # 2 A*YOUR 1.000 0.780
    # 4 B*USER 0.700 1.000
    

    Regex explanation:

    \\b(A\\*YOU|B\\*USE)\\b
    ^^^                 ^^^  "word boundary", meaning the previous/next chars
                             are begin/end of string or from A-Z, a-z, 0-9, or _
       ^               ^     parens "group" the pattern so we can reference it
                             in the replacement string
        ^^^^^^^              literal "A", "*", "Y", "O", "U" (same with other string)
               ^             the "|" means "OR", so either the "A*" or the "B*" strings