地区查询python

def file_to_dict(area_file):
	"""接收一个表示文件名的字符串为参数，将文件中的数据保存到字典中，返回字典。"""
	area_in_dict = {}
	with open(area_file, '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])