思路是,先将需要添加为新的列存储到一个暂时的列表中,然后用到以下函数来存储
data_.loc[:, "新列的名字"] = save_list_
上面的save_list_就是暂时存储了信息的列表了。
以下是我的代码,供以后快速回忆。
schools_data = {
"985": [
"北京师范大学",
"厦门大学",
"华东师范大学",
"中国科学院大学"
],
"211": [
"东北师范大学",
"上海外国语大学",
"中央民族大学",
"华南师范大学",
"湖南师范大学",
"华中师范大学",
"陕西师范大学"
],
"一本": [
"苏州大学",
"天津师范大学",
"深圳大学",
"首都师范大学",
"湖南科技大学",
"广东工业大学",
"暨南大学",
"宁波大学",
"郑州大学",
"河海大学",
"长沙理工大学",
"四川师范大学",
"广西师范大学",
"江西师范大学",
"吉林师范大学",
"广州大学",
"北京语言大学",
"南华大学",
"辽宁大学",
"南通大学",
"西北大学",
"武汉体育学院"
],
"二本": [
"广东技术师范大学",
"广东石油化工学院",
"广州体育学院",
"星海音乐学院",
"广州应用科技学院",
"长沙师范学院",
"南宁师范大学",
"西华师范大学",
"首都体育学院"
],
"港澳": [
"香港大学",
"香港中文大学",
"香港城市大学",
"香港理工大学",
"香港岭南大学"
],
"国外": [
"哥伦比亚大学",
"伦敦大学学院(UCL)",
"悉尼大学",
"墨尔本大学",
"南安普顿大学",
"宁波诺丁汉大学"
]
} # 这个叫deepseek来整理所需要的资料
def cengci_(data_2):
data_ = pd.DataFrame(data_2)
save_list_school = []
for i in data_['毕业院校']:
if i in schools_data['985']:
save_list_school.append('985')
continue
elif i in schools_data['211']:
save_list_school.append('211')
continue
elif i in schools_data['一本']:
save_list_school.append('一本')
continue
elif i in schools_data['二本']:
save_list_school.append('二本')
continue
elif i in schools_data['港澳']:
save_list_school.append('港澳')
continue
elif i in schools_data['国外']:
save_list_school.append('国外')
continue
else:
print(i)
save_list_school.append('未知类别')
continue
data_.loc[:, "毕业院校类别"] = save_list_school # 最主要是这一行,就能添加进去。
print(data_)
data_.to_csv('save.csv', encoding='ANSI')
cengci_(data2)