Site icon R-bloggers

Expand delimited columns in R

[This article was first published on Eldon Prince » R-bloggers, and kindly contributed to R-bloggers]. (You can report issue about the content on this page here)
Want to share your content on R-bloggers? click here if you have a blog, or here if you don't.

A postdoctoral researcher asked me the other day to help him expand a vector of comma delimited values so he could do computations in R with it. I wrote an R function to solve the problem. Here is the before and after:

> data
  Name      Score1   Score2
1 Bill 1,3,4,3,6,9 F1,F3,F2
2  Bob       3,2,3 F2,F2,F4
3  Sam       2,5,3 F5,F2,F4
> expand.delimited(data)
   Name Score1
1  Bill      1
2  Bill      3
3  Bill      4
4  Bill      3
5  Bill      6
6  Bill      9
7   Bob      3
8   Bob      2
9   Bob      3
10  Sam      2
11  Sam      5
12  Sam      3
# Description
# Accepts a data.frame where col1 represents a factor and col2 represents
# comma or other delimited values to be expanded according to col1.
# Returns a data.frame.

# Usage
# expand.delimited(x, ...)

# Default
# expand.delimited(x, col1=1, col2=2, sep=",")

# Arguments
# x     A data.frame
# col1  Column in data.frame to act as factor
# col2  Column in data.frame that is delimited and will be expanded
# sep   Delimiter

#Download data
#Read in data 
data<-read.table("expand_delimited.txt",header=T)

#Function to expand data
expand.delimited <- function(x, col1=1, col2=2, sep=",") {
  rnum <- 1
  expand_row <- function(y) {
    factr <- y[col1]
    strng <- toString(y[col2])
    expand <- strsplit(strng, sep)[[1]]
    num <- length(expand)
    factor <- rep(factr,num)
    return(as.data.frame(cbind(factor,expand),
          row.names=seq(rnum:(rnum+num)-1)))
    rnum <- (rnum+num)-1
  }
  expanded <- apply(x,1,expand_row)
  df <- do.call("rbind", expanded)
  names(df) <- c(names(x)[col1],names(x)[col2])
  return(df)
}

# Example
expand.delimited(data)

To leave a comment for the author, please follow the link and comment on their blog: Eldon Prince » R-bloggers.

R-bloggers.com offers daily e-mail updates about R news and tutorials about learning R and many other topics. Click here if you're looking to post or find an R/data-science job.
Want to share your content on R-bloggers? click here if you have a blog, or here if you don't.