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

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

尝试这个：

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 ]

Answer 5

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