Python是一种高级编程语言,它被广泛应用于各种领域,包括数据分析、Web开发、人工智能等。在日常开发中,经常需要从目录中过滤出指定文件类型的文件,比如只需要查找所有的.jpg文件或者.txt文件。本文将介绍如何使用Python实现这个功能。
一、os模块
Python中的os模块提供了访问操作系统功能的接口,可以用来获取文件信息、目录信息等。在过滤指定文件类型的文件时,需要使用os模块的listdir函数获取目录中的所有文件,然后遍历每个文件,判断其文件类型是否符合要求。
以下是一个简单的示例代码:
import os
def filter_files(path, ext):
files = os.listdir(path)
result = []
for file in files:
if file.endswith(ext):
result.append(file)