Python编程查找文件扩展名的简单方法

在Python编程中，经常会遇到需要查找特定文件扩展名的需求。通过使用Python的内置函数和模块，我们可以轻松地实现这个任务。

方法一：使用glob模块

glob模块是Python自带的一个文件查找模块，它可以根据通配符模式来获取文件列表。以下是一个简单的例子，演示如何使用glob模块来查找指定扩展名的文件：

import glob

def find_files_with_extension(extension):
    file_list = glob.glob('*.' + extension)
    return file_list

以上代码假设你想要查找的文件都在当前工作目录下。find_files_with_extension函数接受一个参数extension，代表你要查找的文件扩展名。函数会返回所有以该扩展名结尾的文件列表。

方法二：使用os模块

另一种常用的方法是使用os模块中的一些函数来获取文件列表，并根据文件名的扩展名进行过滤。以下是一个示例代码：

import os

def find_files_with_extension(extension):
    file_list = [file for file in os.listdir('.') if file.endswith('.' + extension)]
    return file_list

在上述代码中，os.listdir('.')返回当前目录中的所有文件和文件夹的列表。通过使用列表推导式，我们可以过滤并只保留扩展名为extension的文件。

使用示例

extension = 'txt'
files = find_files_with_extension(extension)
for file in files:
    print(file)

运行以上示例代码，将会输出当前目录下所有的扩展名为.txt的文件。

以上两种方法都可以轻松地实现查找文件扩展名的需求。根据具体的情况，你可以选择适合自己的方法来编写代码。希望本文能帮助到你！