Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Mutate changes full column instead of row by row

In a dataframe, I want to create a new column based on the occurence of a specific set of strings (char vector) in another column.

So basically, I want this:

ID  Phrases
1   some words
2   some words dec
3   some words nov may

to return this:

ID  Phrases             MonthsOccur
1   some words          NA
2   some words dec      dec
3   some words nov may  may nov

I have tried the following, and I'm not sure why it's giving me the outcome that it does:

library(dplyr)

vMonths <- c("jan","feb","mar","apr","may","jun","jul","aug","sept","nov","dec")

a <- c(1,2,3)
b <- c('phrase number one', 'phrase dec','phrase nov')

df <- data.frame(a,b)
names(df) <- c("ID","Phrases")
df <- df %>% mutate(MonthsOccur = paste(vMonths[str_detect(Phrases, vMonths)],collapse=" "))

It gives me the following warning:

Warning message: In stri_detect_regex(string, pattern, negate = negate, opts_regex = opts(pattern)) : longer object length is not a multiple of shorter object length

And the following outcome:

ID  Phrases             MonthsOccur
1   some words          dec
2   some words dec      dec
3   some words nov may  dec
like image 862
BroQ Avatar asked Oct 14 '25 07:10

BroQ


1 Answers

One option is to apply str_detect rowwise

library(dplyr)
library(stringr)

df %>%
  rowwise() %>%
  mutate(MonthsOccur = paste0(vMonths[str_detect(Phrases, vMonths)], 
                       collapse = " "))

However, rowwise may or may not be continued in the future so a better way is to use map operations

df %>%
  mutate(MonthsOccur = purrr::map_chr(Phrases,  
                      ~paste0(vMonths[str_detect(.x, vMonths)], collapse = " ")))

#  ID           Phrases MonthsOccur
#1  1 phrase number one            
#2  2        phrase dec         dec
#3  3    phrase nov may     may nov

A base R option would be with regmatches and gregexpr

sapply(regmatches(df$Phrases, gregexpr(paste0(vMonths, collapse = "|"),
        df$Phrases)), paste0, collapse = " ")

data

df <- structure(list(ID = c(1, 2, 3), Phrases = structure(c(3L, 1L, 
2L), .Label = c("phrase dec", "phrase nov may", "phrase number one"
), class = "factor")), class = "data.frame", row.names = c(NA, -3L))
like image 83
Ronak Shah Avatar answered Oct 18 '25 06:10

Ronak Shah