函数错误：get() 不能应用于字符串（python）

Question

假设我的数据集如下所示：

ID	学校名称	学校类型	班级人数	专业化
0	新泽西州立学校	公立学校	7000	一般的
1	美国国际学校	私立国际学校	200	干
2	波士顿艺术高中	私立国立学校	500	艺术
3	新汉普顿学院	私立国立学校	300	人文学科
4	科罗拉多州立学校	公立学校	10000	一般的
...	...	...	...	...

我定义了一些关键值：

school_type_key={'State School':0, 'Private National School':1, 'Private International School':2}
specialization_key={'STEM':0,'Humanities':1,'Arts':2, 'Life Sciences':3, 'General':4, 'Other':5}

我需要编写一个函数：

1. 占一列
2. 列出唯一值
3. 根据定义的键对该列表进行排序
4. 返回一个列表

这是我尝试过的：

def sort_by_keys(column):
    unique_values=df[column].unique()
    unique_values=list(unique_values)
    key=column.replace(' ','_').lower()+'_key'

    #I need key so that if i pass 'School Type' column, it is sorted by 'school_type_key'

    unique_values.sort(key=key.get, reverse=True)
    return unique_values

sort_by_keys('School Type')

但是，它返回一个错误，因为 key 是一个字符串，并且 'get' 不能应用于字符串。 我该如何解决这个问题？

预期输出：

output=sort_by_keys('School Type')
print(output)
['State School', 'Private National School', 'Private International School']

output=sort_by_keys('Specialization')
print(output)
['STEM', 'Humanities','Arts','General']

Answer 1

您可以使用：

MAPPING = {
    'School Type': school_type_key,
    'Specialization': specialization_key
}

def sort_by_keys(col):
    return (df[col].sort_values(key=lambda x: x.map(MAPPING[col]))
                   .unique().tolist())

输出：

>>> sort_by_keys('School Type')
['State School', 'Private National School', 'Private International School']

>>> sort_by_keys('Specialization')
['STEM', 'Humanities', 'Arts', 'General']

Answer 2

我不完全了解您要达到的目标。 但是根据您的字典school_type_key ，这可能对您有用。：

import pandas as pd
school_type_key={'State School':0, 'Private National School':1, 'Private International School':2}
df = pd.DataFrame({'School Type': school_type_key.keys()})
def sort_by_keys(column):
    return df[column].to_list()[::-1]
sort_by_keys('School Type')

结果：

['Private International School', 'Private National School', 'State School']