Question

I have a data.frame where the data entries are entered in this format 1,2,3,10. That is, they are comma separated integers that range from 0-20, and do not need to be consecutive. Each is currently considered a factor. I have four variables that contain these values, and I'd like to create a new variable, that includes a given integer only if it is in three of the the four variables, if there are not three occurrences of an integer, then use 0.

M1    M2      M3      M4      M_NEW
1     1,2     0        1       1
3,4   3,4   1,2,3,4    4       3,4

I am unsure on how to deal with these comma separated integers. If they were single integers, I could do something like this:

# modified from https://stackoverflow.com/a/14114085/1670053
# over each row of data.frame (or matrix)
sapply(1:nrow(df), function(idx) {
# get the number of time each entry in df occurs
    t <- table(t(g[idx, ]))
# get the maximum count (or frequency)
    if(max(t) > 2){
      t.max <- max(t)
      }else{ t.max <- 0
    }
# get all values that equate to maximum count
    t <- as.numeric(names(t[t == t.max]))

})

Though with these multiple values separated by commas, I am unsure where to start.

# data and example output
df <- structure(list(M1 = structure(c(3L, 2L, 2L, 5L, 3L, 1L, 7L, 1L, 
8L, 1L, 3L, 4L, 3L, 6L), .Label = c("0", "1", "1,2", "1,2,3", 
"1,2,3,4", "1,2,3,4,5", "3,4,5,6,7", "6,7,8,9,10,11,12,13,14,15,16"
), class = "factor"), M2 = structure(c(5L, 2L, 2L, 4L, 4L, 1L, 
11L, 8L, 7L, 9L, 3L, 6L, 3L, 10L), .Label = c("0", "1,2", "1,2,3", 
"1,2,3,4", "1,2,3,4,5", "1,2,3,4,5,6,7", "1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16", 
"2", "2,3,4,5", "4,5,6", "4,5,6,7,8,9,10,11,12,13,14"), class = "factor"), 
    M3 = structure(c(4L, 1L, 1L, 8L, 3L, 1L, 6L, 1L, 7L, 3L, 
    2L, 5L, 9L, 3L), .Label = c("0", "1,2", "1,2,3,4", "1,2,3,4,5", 
    "1,2,3,4,5,6", "1,2,3,4,5,6,7,8", "1,2,3,4,5,6,7,8,9,10,11,12,13,14", 
    "3,4", "3,4,5"), class = "factor"), M4 = structure(c(5L, 
    1L, 2L, 8L, 2L, 1L, 6L, 3L, 4L, 1L, 3L, 3L, 7L, 9L), .Label = c("0", 
    "1", "1,2", "1,2,3,4,5,12,13,14,15,16,17", "1,2,3,4,5,6", 
    "1,2,3,4,5,6,7,8,9,10,11,12", "3,4", "4", "4,5"), class = "factor"), 
    M_NEW = structure(c(6L, 1L, 2L, 8L, 3L, 1L, 9L, 1L, 7L, 1L, 
    3L, 4L, 5L, 10L), .Label = c("0", "1", "1,2", "1,2,3", "1,2,3,", 
    "1,2,3,4,5", "1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16", "3,4", 
    "3,4,5,6,7,8", "4,5"), class = "factor")), .Names = c("M1", 
"M2", "M3", "M4", "M_NEW"), class = "data.frame", row.names = c(NA, 
-14L))
Was it helpful?

Solution

f <- function(x, n=3) {
  tab <- table(strsplit(paste(x, collapse=","), ","))
  res <- paste(names(tab[which(tab >= n)]), collapse=",")
  return(ifelse(res == "", "0", res))
}
(df[, 5] <- apply(df[, 1:4], 1, f))
# [1] "1,2,3,4,5"                             
# [2] "0"                                     
# [3] "1"                                     
# [4] "3,4"                                   
# [5] "1,2"                                   
# [6] "0"                                     
# [7] "3,4,5,6,7,8"                           
# [8] "0"                                     
# [9] "1,10,11,12,13,14,15,16,2,3,4,5,6,7,8,9"
# [10] "0"                                     
# [11] "1,2"                                   
# [12] "1,2,3"                                 
# [13] "3"                                     
# [14] "4,5"  
Licensed under: CC-BY-SA with attribution
Not affiliated with StackOverflow
scroll top