存储int显示标签?枚举

时间:2017-04-04 20:56:55

标签: python pandas enums

pandas数据框中是否有一种方法可以将数据作为整数存储在列中,但是当我在屏幕上将其打印出来以显示相应的标签时。

类似于Enum map int< =>标签

1 个答案:

答案 0 :(得分:1)

分类数据可以完成这项工作。感谢

 df = pd.DataFrame({"A" : ['c', 'b','f', 'e']})
 df.A.astype('category')
 df.A.cat.categories
 Index([u'b', u'c', u'e', u'f'], dtype='object')

 df.A.cat.codes

 0    1
 1    0
 2    3
 3    2
 dtype: int8

memory usage