Python - 列出带有扩展名的目录中的文件

在 Python 中，我们可以使用 glob 模块来列出一个目录中所有带有指定扩展名的文件。

代码示例

import glob

# 设置要列出的目录和扩展名
dir_path = "/path/to/directory"
extension = "*.txt"

# 通过 glob.glob() 函数取得指定目录下的所有文件名
files = glob.glob(dir_path + extension)

# 输出文件名列表
print(files)

代码解析

1. 导入 glob 模块。
2. 设置要列出的目录路径和扩展名。
3. 通过 glob.glob() 函数获取指定目录下所有扩展名为 extension 的文件名，保存在 files 变量中。
4. 输出 files 变量的内容，即所有符合条件的文件名列表。

注意事项

• 在设置 dir_path 变量的值时需要注意路径的书写格式，不同操作系统可能有差异。
• 扩展名需要包含 * 通配符，如 *.txt。
• 如果不指定扩展名，则会返回目录下所有文件的文件名。例如 files = glob.glob(dir_path)。

总结

使用 glob 模块可以很方便地列出一个目录中符合条件的文件名列表。请注意在使用时注意参数的书写格式和扩展名的通配符。