PYTHON地区查询
任务描述
附件文件中包含全国各地区的地区编码，对应身份证的前6位。身份证的前2位代表省名，在文件中可能存在身份证前2位数字后面加4个0的编码，代表省、自治区或直辖市；身份证号前4位代码地区名，在文件中可能存在身份证前4个数字后面加2个0的编码，表示市或地区。省和地区这两个编码也有可能不存在。输入身份证号前6位，查询到该编码对应的省、市、县（区）。某一项查询不到时，返回并输出空字符串。
输入格式
输入身份证号前6位
输出格式
参考示例分三行输出省、市、县（区）
示例 1
输入： 110113 输出： 北京 顺义区
示例 2
输入： 421221 输出： 湖北 咸宁市 嘉鱼县 平均代码量 15 行
开始你的任务吧，祝你成功！
def file_to_dict(area_file):
"""接收一个表示文件名的字符串为参数，将文件中的数据保存到字典中，返回字典。"""
area_in_dict = {}
with open('IDcode.txt', 'r', encoding='utf-8') as data: # 读文件
for x in data: # 遍历文件对象
ls = x.strip().split(',') # 每行数据切分为列表
area_in_dict[ls[0]] = ls[1] # 得到保存地区编码的字典
return area_in_dict # 返回地区编码字典
def query_area(area_id, area):
"""接收身份证号前6位和编码字典为参数，根据身份证前6位的编码查询所在的省、市、区，以元组形式返回。"""
province = area.get(area_id[:2] + '0000', '') # 省、直辖市和自治区
city = area.get(area_id[:4] + '00', '') # 市或地区
area = area.get(area_id) # 县或区
return province, city, area
if __name__ == '__main__':
areaFile = '8.1 IDcode.txt'
areaId = input()
area_dict = file_to_dict(areaFile) # 调用函数读文件返回地区码字典
address = query_area(areaId, area_dict) # 调用函数查询省、市和区
print(address[0]) # 输出
print(address[1])
print(address[2])