Basic Python Dataframe Sorting Hat Test Pottermore