Python怎么获取当前目录指定扩展名的文件名

Python怎么获取当前目录指定扩展名的文件名

Python编写程序获取当前目录下指定扩展名的文件名，可以通过内置的os和os.path模块实现。首先，你需要输入你感兴趣的文件扩展名，例如.py。接着，程序会遍历当前目录中的所有文件，筛选出具有指定扩展名的文件，并打印它们的名称。具体代码如下：

import os
import os.path
ext = input('请输入你要显示的文件的扩展名（如.py）：')
print([f for f in os.listdir() if os.path.isfile(f) and os.path.splitext(f)[1] == ext])

这段代码的工作原理是，os.listdir()函数返回当前目录下的所有文件和目录名。然后，os.path.isfile(f)检查f是否为文件，os.path.splitext(f)[1]返回文件扩展名。只有当文件扩展名与输入的ext匹配时，该文件才会被加入到列表中并打印。

例如，如果你输入.py作为扩展名，程序将列出所有扩展名为.py的文件。这个过程简单高效，适用于需要快速查看特定类型文件的场景。使用此代码，你可以轻松地获取当前目录下所有指定扩展名的文件名，便于进一步处理或分析这些文件。

尽管此代码段简洁，但它提供了一个强大的工具，可以方便地处理文件系统中的数据。通过修改代码中的输入提示和输出方式，可以扩展其功能以满足不同的需求。例如，你可以添加更多的输入验证，或者将输出存储到文件中，以便日后查阅。2024-12-21