python查找文件夹下所有路径的多种特定后缀文件

原创

mob649e81680b4f 2024-02-04 05:51:48 ©著作权

©著作权归作者所有：来自51CTO博客作者mob649e81680b4f的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python查找文件夹下所有路径的多种特定后缀文件

作为一名经验丰富的开发者，你可以帮助这位刚入行的小白实现Python查找文件夹下所有路径的多种特定后缀文件的功能。下面是整个流程的步骤：

流程图

flowchart TD
    A[开始] --> B[指定文件夹路径]
    B --> C[获取文件夹下的所有文件和文件夹]
    C --> D[遍历文件夹下的所有文件和文件夹]
    D --> E[判断是否是文件]
    E --> F[判断文件后缀是否满足条件]
    F --> G[将满足条件的文件路径添加到结果列表]
    G --> D
    D --> H[判断是否是文件夹]
    H --> C
    C --> I[返回结果列表]
    I --> J[结束]

代码实现步骤

指定文件夹路径：首先，我们需要指定要查找的文件夹路径。可以使用os模块的path方法来获取当前文件夹的绝对路径，或者指定其他文件夹路径。
```
import os

folder_path = os.path.abspath(".") # 获取当前文件夹的绝对路径
```
获取文件夹下的所有文件和文件夹：使用os模块的listdir方法获取指定文件夹下的所有文件和文件夹的名称。
```
file_list = os.listdir(folder_path) # 获取文件夹下的所有文件和文件夹
```
遍历文件夹下的所有文件和文件夹：使用for循环遍历文件夹下的所有文件和文件夹。
```
for file_name in file_list:
    # 进行后续操作
```

判断是否是文件：使用os模块的path方法的isfile函数来判断当前处理的对象是否是文件。

file_path = os.path.join(folder_path, file_name) # 构建文件的完整路径
if os.path.isfile(file_path):
    # 进行后续操作

判断文件后缀是否满足条件：使用os模块的path方法的splitext函数来获取文件的后缀名，然后与特定的后缀进行比较。
```
file_ext = os.path.splitext(file_name)[1] # 获取文件的后缀名
if file_ext in [".txt", ".doc"]:
    # 进行后续操作
```
将满足条件的文件路径添加到结果列表：使用一个列表来保存满足条件的文件路径。
```
result_list = [] # 结果列表
result_list.append(file_path) # 添加满足条件的文件路径
```
判断是否是文件夹：使用os模块的path方法的isdir函数来判断当前处理的对象是否是文件夹。
```
if os.path.isdir(file_path):
    # 进行后续操作
```
返回结果列表：在遍历结束后，返回结果列表。
```
return result_list
```

完整代码示例

import os

def find_files(folder_path, extensions):
    result_list = [] # 结果列表

    # 获取文件夹下的所有文件和文件夹
    file_list = os.listdir(folder_path)
    
    # 遍历文件夹下的所有文件和文件夹
    for file_name in file_list:
        file_path = os.path.join(folder_path, file_name) # 构建文件的完整路径
        
        # 判断是否是文件
        if os.path.isfile(file_path):
            # 判断文件后缀是否满足条件
            file_ext = os.path.splitext(file_name)[1] # 获取文件的后缀名
            if file_ext in extensions:
                # 将满足条件的文件路径添加到结果列表
                result_list.append(file_path)
        
        # 判断是否是文件夹
        if os.path.isdir(file_path):
            # 递归调用查找文件的函数
            result_list.extend(find_files(file_path, extensions))
    
    # 返回结果列表
    return result_list

folder_path = os.path.abspath(".") # 获取当前文件夹的绝