Python进行数据处理练习题之行政区划分

编程要求

test文件夹下行政区划分.txt文件如下图所示，每行有代码和行政区域名称两列

其中，省级行政区划代码的后四位是0000。编写程序，将文件按省级行政区划分为多个文本文件，文件名为行政区域名称，例如天津市、河北省、澳门特别行政区等，存储在test的dq文件夹下，文件列表如下所示

在新文件中，存储该行政区的所有编码和地区。
没有空行，且每行去除首尾的空格。
如下图所示

参考答案

import shutil
shutil.rmtree("test/dq")
import os
os.mkdir("test/dq")
f1=open("test/行政区划.txt","r",encoding="utf8")
#代码开始
citys = []
urban = []
for i in f1:
    if "0000" in i:
        citys.append(urban.copy())
        urban.clear()
        urban.append(i.strip())
    else:
        urban.append(i.strip())
for city in citys[1:]:
    with open(f"test/dq/{city[0][6:]}.txt", "w") as f:
        f.write("\n".join(city))
#代码结束
f1.close()