从Python中的行值中提取列

时间:2015-03-24 13:45:29

标签: pandas pivot

我正在使用此数据框:

dfPredET.head(5)

id  Class
1   Class_2
2   Class_1
3   Class_6
4   Class_2
5   Class_1

我想转换它,表明一个实例是属于类(1)还是不属于(0):

id  Class_1 Class_2 Class_3 Class_4 Class_5 Class_6 Class_7 Class_8 Class_9
1   0   1   0   0   0   0   0   0   0
2   1   0   0   0   0   0   0   0   0
3   0   0   0   0   0   1   0   0   0
4   0   1   0   0   0   0   0   0   0
5   1   0   0   0   0   0   0   0   0

我可以使用pivot()功能吗?怎么样?

1 个答案:

答案 0 :(得分:2)

使用get_dummies

In [7]:

pd.get_dummies(df)
Out[7]:
   id  Class_Class_1  Class_Class_2  Class_Class_6
0   1              0              1              0
1   2              1              0              0
2   3              0              0              1
3   4              0              1              0
4   5              1              0              0