handling categorical data in python