Question

是否可以使用os.listdir命令查看具有特定扩展名的文件？我希望它能够工作，因此最终可能只显示带有.f的文件或文件夹。我检查了文档，什么也没找到，所以不要问。

Answer 1

glob擅长：

import glob
for f in glob.glob("*.f"):
    print f

Answer 2

不要问什么？

[s for s in os.listdir() if s.endswith('.f')]

如果要查看扩展名列表，可以进行明显的概括，

[s for s in os.listdir() if s.endswith('.f') or s.endswith('.c') or s.endswith('.z')]

或其他方式写一点：

[s for s in os.listdir() if s.rpartition('.')[2] in ('f','c','z')]

Answer 3

到目前为止还没有提到另一种可能性：

import fnmatch
import os

for file in os.listdir('.'):
    if fnmatch.fnmatch(file, '*.f'):
        print file

实际上这就是glob模块的实现方式，因此在这种情况下glob更简单，更好，但fnmatch模块在其他情况下可以派上用场，例如使用os.walk进行树遍历时。

Answer 4

[s for s in os.listdir() if os.path.splitext(s) == 'f']

Answer 5

试试这个：

from os import listdir

extension = '.wantedExtension'

mypath = r'my\path'

filesWithExtension = [ f for f in listdir(mypath) if f[(len(f) - len(extension)):len(f)].find(extension)>=0 ]

Python：os.listdir替代/某些扩展

5 个答案: