听说你文件夹非常难找？用Python打造一款文件搜索工具，所有功能自己定义！

本文链接：https://blog.csdn.net/pythonjavaphp/article/details/129186422

一、前言

大家好，在日常的办公中，我们经常会从一堆不同格式的文件(夹)中搜索特定的文件，可能你是凭着记忆去找或是借助软件，但你有想过如何用Python实现吗？本文将基于几个常见的搜索操作讲解

二、扫描路径内的内容

有些时候我们会希望在当前文件夹的成百上千个文件中快速找到需要的文件，如果这个文件夹又包括很多个子文件夹，并不需要程序进去查找而无端耗费资源。这就是典型的「扫描一层搜索」

1. 基于 os.scandir()

需求说明

❝

输出C:\Program Files (x86)路径下名字包含“Windows”的文件夹名称，并统计个数

❞

步骤分析

这是一个非遍历的需求，只需要在目的文件夹内扫描一圈判断即可。用到的方法是os.scandir()，使用如下：

import os

path = ...
for file in os.scandir(path):
print(file.name, file.path, file.is_dir())

上面代码最后输出的是给定路径下各内容的名字、绝对路径，并判断其是否是文件夹

需要注意的是os.scandir()只在路径下一级扫描，需求实现的代码如下(代码逻辑很简单，可以自己做相应调整)：

import os

num = 0
for file in os.scandir(r'C:\\Program Files (x86)'):
    if file.is_dir():
        if 'Windows' in file.name:
            print(file.name)
            num += 1
print('含有Windows的文件夹个数为：', num)

2. 基于 os.listdir()

需求说明

❝

输出C:\Program Files (x86)路径下所有可执行文件 (即后缀为 .exe)

❞

步骤分析

依然是非遍历的需求，这里使用os.listdir()，它比os.scandir()简单一些，可直接调用输出名称而非路径。后缀名可以用字符串的切片来判断，但难免失去灵活性。此时建议用字符串方法string.endswith()来判断名称的结尾是否是.exe，代码如下：

import os

for file in os.listdir(r'C:\\Program Files (x86)'):
if file.endswith('.exe'):
print(file)

三、遍历文件夹搜索文件

更多时候我们希望给定一个大概的路径，在这个路径下的所有文件夹里一层一层找，找到特定文件或者符合要求的文件，这里需要遍历文件，有两种主要的方法：

1. 基于 os.walk()

os.walk 遍历后产生三个参数：当前文件夹路径, 包含文件夹名称 [列表形式], 包含文件名称[列表形式]

可以用如下代码完成简单遍历：

import os

for dirpath, dirnames, filenames in os.walk(r'C:\\Program Files (x86)'):
    print(f'打开文件夹{dirpath}')
    if dirnames:
        print(dirnames)
    if filenames:
        print(filenames)
    print('-' * 10)

需求说明

❝

遍历C:\Program Files (x86)找出所有新版Excel文件 (即后缀为.xlsx)

❞

步骤分析

只要理解了os.walk()的工作模式，用endswith()判断后缀即可。最后如果需要获得绝对路径可以把当前文件夹路径和文件名拼接，简单一点用 + 或者字符串格式化，也可以用os模块内的方法

import os

for dirpath, dirnames, filenames in os.walk(r'C:\\Program Files (x86)'):
    if filenames:
        for i in filenames:
            if i.endswith('.xlsx'):
                print(os.path.join(dirpath, i))