如何将二进制变量的DataFrame列转换为多个虚拟变量列

时间:2017-04-21 15:59:30

标签: python pandas binary dummy-variable

这应该是一个简单的问题,但由于某种原因,我无法在线找到答案。我有一个由虚拟变量组成的DataFrame列:

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

我想要的是一个看起来像

的4x4数据框
A B C D
0 1 1 0
0 1 1 1
1 0 0 0
0 0 1 1

我尝试过使用get_dummies没有结果:

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并将列放入一系列列表中也不起作用。我该怎么办?

3 个答案:

答案 0 :(得分:3)

你可以试试这个:

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

enter image description here

答案 1 :(得分:2)

这会跳过foofoo1的初始步骤,然后从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

答案 2 :(得分: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