Basic Python Dataframe Sorting Hat Test Official