How to turn a DataFrame column of binary variables into multiple columns of dummy variables

Question

This should be an easy question, but for some reason I can't find the answer online. I have a DataFrame column that consists of dummy variables:

import pandas as pd

foo = pd.Series([6,7,8,3])
foo1 = bob.apply(lambda x: bin(x)[2:].zfill(4))
foo1

0    0110
1    0111
2    1000
3    0011

What I want is a 4x4 data frame that looks like

I've tried using get_dummies to no results:

foo1.str.get_dummies()

0110 0111 1000 0011
1    0    0    0
0    1    0    0
0    0    1    0
0    0    0    1

str.split and making the column into a series of lists doesn't work either. What should I do?

akuiper · Accepted Answer · 2017-04-21 16:06:58Z

3

You can try this:

# convert the series to str type; 
# extract all characters with regex .; 
# unstack to wide format
foo1.astype(str).str.extractall('(.)')[0].unstack()

answered Apr 21, 2017 at 16:06

akuiper

216k33 gold badges362 silver badges379 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

piRSquared · Accepted Answer · 2017-04-21 17:32:15Z

2

This will skip your initial step of foo to foo1 and get you straight there from foo

foo.apply(lambda x: pd.Series(list('{:04b}'.format(x))))

   0  1  2  3
0  0  1  1  0
1  0  1  1  1
2  1  0  0  0
3  0  0  1  1

answered Apr 21, 2017 at 17:32

piRSquared

296k68 gold badges509 silver badges654 bronze badges

Comments

MaxU - stand with Ukraine · Accepted Answer · 2017-04-21 17:32:50Z

2

In [49]: pd.DataFrame(foo1.apply(list).values.tolist())
Out[49]:
   0  1  2  3
0  0  1  1  0
1  0  1  1  1
2  1  0  0  0
3  0  0  1  1

answered Apr 21, 2017 at 17:32

MaxU - stand with Ukraine

212k37 gold badges402 silver badges436 bronze badges

Collectives™ on Stack Overflow

How to turn a DataFrame column of binary variables into multiple columns of dummy variables

3 Answers 3

Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related