导航

pandas函数get_dummies的坑

Posted on 2018-10-29 14:34  wzd321  阅读(2514)  评论(0编辑  收藏  举报

转载:https://blog.csdn.net/mvpboss1004/article/details/79188190

pandas中的get_dummies得到的one-hot编码数据类型是uint8,进行数值计算时会溢出!!!

import pandas as pd
import numpy as np
a = [1, 2, 3, 1]
one_hot = pd.get_dummies(a)
print(one_hot.dtypes)
print(one_hot)
print(-one_hot)

1 uint8
2 uint8
3 uint8
dtype: object


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


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

one_hot = one_hot.astype('float') print(-one_hot)

1 2 3

0 -1.0 -0.0 -0.0

1 -0.0 -1.0 -0.0

2 -0.0 -0.0 -1.0

3 -1.0 -0.0 -0.0