使用 os.path 基于扩展名搜索文件

Searching files based on extension using os.path

我想知道如何根据扩展名搜索文件。如果你要搜索 py 或 txt 或其他一些扩展名,用户会输入它然后它会搜索它。我不允许使用 glob 或 os.walk

我有这个方法。如有必要,我会 post 整个代码。

用户先输入一个目录,然后是另一种方法,他们会说

E txt 或 E .txt(均有效),它会 return 根据您使用的扩展名生成文件。不过我不确定我的代码有什么问题。

在您的代码中,

 for file in directory 

肯定是错误的,因为 directory 是一个包含路径的字符串。应该是

for file in os.listdir(directory):

获取目录中的文件列表