import pandas as pd #设置路径 os.chdir(r"C:\\Users\\Desktop\\论坛") os.getcwd() #导入数据 data = pd.read_csv("data.csv",engine='python') city= open("城市字典库.txt",encoding="UTF-8") data.head(7)
#生成城市字典库 city1=city.split("\n")#将其变为列表 city1 del city1[0]#删除第一个元素的空格 city1
#将txt变成字典 city_dic={} for row in city1: cit=row.split(":")[0] pro=row.split(":")[1] city_dic[eval(cit)]=pro.strip('",')#去掉引号和逗号 city_dic
#map函数是一个映射
data['province1'] =data.iloc[:,6].map(city_dic) data
可以看到省份已经完成了匹配,如果需要将城市与几线城市进行匹配,也可以同样导入城市与几线排名的字典库。