Using Pandas for features categorization