实现功能
依次遍历每一行，在某列包含某个元素时新增一列进行标记
实现代码
def province_distribution_of_colleges(self, file):
df = pd.read_excel(os.path.join(self.datapath, file))
df1 = df
hua_bei = ['北京市','天津市','河北省','山西省','内蒙古自治区']
dong_bei = ['辽宁省','吉林省','黑龙江省']
hua_dong = ['上海市','江苏省','浙江省','安徽省','福建省','江西省','山东省']
hua_nan = ['广东省','广西壮族自治区','海南省']
zhong_nan = ['湖南省','湖北省','河南省','江西省']
xi_nan = ['重庆市','四川省','贵州省','云南省','西藏自治区']
xi_bei = ['陕西省','甘肃省','青海省','宁夏回族自治区','新疆维吾尔自治区']
gang_ao = ['香港特别行政区','澳门特别行政区']
df1['区域'] = None
for index, row in df1.iterrows():
if row['省份'] in hua_bei:
df1.at[index, '区域'] = '华北'
elif row['省份'] in dong_bei:
df1.at[index, '区域'] = '东北'
elif row['省份'] in hua_dong:
df1.at[index, '区域'] = '华东'
elif row['省份'] in hua_nan:
df1.at[index, '区域'] = '华南'
elif row['省份'] in zhong_nan:
df1.at[index, '区域'] = '中南'
elif row['省份'] in xi_nan:
df1.at[index, '区域'] = '西南'
elif row['省份'] in xi_bei:
df1.at[index, '区域'] = '西北'
elif row['省份'] in gang_ao:
df1.at[index, '区域'] = '港澳'
else:
df1.at[index, '区域'] = '未知'
print(df1)
province_distribution_of_colleges('schools_with_coordinates.xlsx')实现效果
本人读研期间发表5篇SCI数据挖掘相关论文，现在某研究院从事数据挖掘相关科研工作，对数据挖掘有一定认知和理解，会结合自身科研实践经历不定期分享关于python机器学习、深度学习、数据挖掘基础知识与案例。
致力于只做原创，以最简单的方式理解和学习，关注我一起交流成长。
邀请三个朋友关注V订阅号：数据杂坛，即可在后台联系我获取相关数据集和源码，送有关数据分析、数据挖掘、机器学习、深度学习相关的电子书籍。
