将 Pandas 中的文本转换为 int64 分类

Question

我有一些艺术家姓名，data['artist']我想通过以下方式转换为分类列：

x = data['artist'].astype('category').cat.codes
x.dtype 

返回：

dtype('int32')

我得到负数，这表明存在某种溢出情况。因此，我想使用np.int64它，但我找不到有关如何完成此操作的文档。

x = data['artist'].astype('category').cat.codes.astype(np.int64)
x.dtype

给予

dtype('int64')

但很明显 int32 已转换为 int64，因此负值仍然存在

x = data['artist'].astype('category').cat.codes.astype(np.int64)
x.min()

-1

Answer 1

我认为你NaN在专栏中有artist，所以代码是-1：

data=pd.DataFrame({'artist':[np.nan,'y','z','x','y','z']})

x = data['artist'].astype('category').cat.codes
print x
0   -1
1    1
2    2
3    0
4    1
5    2
dtype: int8

为了检查NaN你可以使用isnull：

print data[data.artist.isnull()]
  artist
0    NaN