結果 : how to categorize data in python