Convert a factor column to multiple boolean columns

Question

Given data that looks like:

library(data.table)
DT <- data.table(x=rep(1:5, 2))

I would like to split this data into 5 boolean columns that indicate the presence of each number.

I can do this like this:

new.names <- sort(unique(DT$x))

DT[, paste0('col', new.names) := lapply(new.names, function(i) DT$x==i), with=FALSE]

But this uses a pesky lapply which is probably slower than the data.table alternative and this solutions strikes me as not very "data.table-ish".

Is there a better and/or faster way to create these new columns?

Would something like model.matrix be helpful? model.matrix(~cols-1) — BenBarnes
– BenBarnes, Commented Jul 5, 2012 at 18:51

BenBarnes · Accepted Answer · 2012-07-05 19:00:10Z

How about model.matrix?

model.matrix(~factor(x)-1,data=DT)

   factor(x)1 factor(x)2 factor(x)3 factor(x)4 factor(x)5
1           1          0          0          0          0
2           0          1          0          0          0
3           0          0          1          0          0
4           0          0          0          1          0
5           0          0          0          0          1
6           1          0          0          0          0
7           0          1          0          0          0
8           0          0          1          0          0
9           0          0          0          1          0
10          0          0          0          0          1
attr(,"assign")
[1] 1 1 1 1 1
attr(,"contrasts")
attr(,"contrasts")$`factor(x)`
[1] "contr.treatment"

Apparently, you can put model.matrix into [.data.table to give the same results. Not sure if it would be faster:

DT[,model.matrix(~factor(x)-1)]

mnel · Accepted Answer · 2012-07-05 23:28:04Z

2

There is also nnet::class.ind

library(nnet)

cbind(DT, setnames(as.data.table(DT[, class.ind(x)]),paste0('col', unique(DT$x))))

answered Jul 5, 2012 at 23:28

mnel

116k28 gold badges269 silver badges255 bronze badges

Comments

Gennady Khvorykh · Accepted Answer · 2016-05-03 03:15:19Z

0

library(data.table)
DT <- data.table(x=rep(1:5, 2))

# add column with id
DT[, id := seq.int(nrow(DT))]

# cast long table into wide
DT.wide <- dcast(DT, id ~ x, value.var = "x", fill = 0, fun = function(x) 1)

answered May 3, 2016 at 3:15

Gennady Khvorykh

1

Collectives™ on Stack Overflow

Convert a factor column to multiple boolean columns

3 Answers 3

Comments

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related