Python在日常中的使用

数据处理与分析

发布于 2019-07-31 08:28:55

9.9K00

代码可运行

文章被收录于专栏：数据处理与分析数据处理与分析

运行总次数：0

代码可运行

01—问题

今天想要整理下电脑硬盘的文件,只要一些有用的方便共享，然后发现文件组织结构是这个样子的

而我只想保留其中的压缩包，怎么办？手动删除吗？这不符合咱一贯的行事风格啊。毕竟，能动脑的，就不要动手，接下来就随我一起，干掉这些多余文件吧！

02—解决问题

人

生

苦

短

直接上代码截图吧，可以有一个直观的了解，由于代码比较简单，所以就不再赘述。

如果感觉需要进行进一步对代码进行阐述，欢迎在下方投票区进行投票，以便于我能了解大家的需求，写出大家愿意看的文字。

import os
import re
from shutil import rmtree

#构建正则表达式
#在具体使用中需要根据实际情况调整表达式
pattern1 = re.compile('.+兴趣班第\d{1,2}期$')
pattern2 = re.compile('.+\.rar$|.+\.zip$|.+\.7z$')

#定义一个函数，参数就用起始路径

def func_wj(fpath):
   for lj,_,wjs in os.walk(fpath):
        if pattern1.match(lj) or lj==fpath:
            for wj in wjs:
                if pattern2.match(wj):
                    pass
               else:
                   os.remove(os.path.join(lj,wj))
        else:
           rmtree(lj)
if __name__ == '__main__':
        
   func_wj('这里换成你自己的路径')