Here you will find daily news and tutorials about R, contributed by over 750 bloggers.
There are many ways to follow us - By e-mail:On Facebook: If you are an R blogger yourself you are invited to add your own R content feed to this site (Non-English R bloggers should add themselves- here)

Most R functions used in econometrics convert factor variables into a set of dummy/binary variables automatically. This is useful when estimating a linear model, saving the user from the laborious activity of manually including the dummy variables as regressors. However, what if you want to reshape your dataframe so that it contains such dummy variables?

The following function, datdum(.), is a simple workaround. The first argument is the factor variable (which can also be a character), the second is the dataframe and the third is the name you want to call these dummy variables.

datdum <- function(x, data, name){
data$rv <- rnorm(dim(data)[1],1,1)
mm <- data.frame(model.matrix(lm(data$rv~-1+factor(data[,x]))))
names(mm) <- paste(name,1:dim(mm)[2],sep=".")
data$rv <- NULL
data <- cbind(data,mm)
return(data)
}
# simple example
dat <- c("A","B","C")
dat <- data.frame(dat)
datdum(x="dat",data=dat,name="category")
#########################
# output
#########################
> dat
dat
1 A
2 B
3 C
> datdum(x="dat",data=dat,name="category")
dat category.1 category.2 category.3
1 A 1 0 0
2 B 0 1 0
3 C 0 0 1

Related

To leave a comment for the author, please follow the link and comment on their blog: DiffusePrioR » R.