Using pyton how to assign each row of a dataset to a category so each row is in one category

Question

I\'m not sure how to explain this in a phytonic friendly way but I would like to know if its possible to use python to assign each row in a dataset to a group/category but strat